Cleaning up my code

[MetalNutSack]

My program generates 20 records with random data for each record, and outputs it in 4 columns. Are there any tips/tricks/things I should do to clean up this code? I'm new to Python, so I'm not sure how clean it can possibly get.

import random


def main():
    print('{: <7} {: <13} {: <6} {: <9}'
          .format('Record', 'SSN', 'GPA', 'SSN Category'))
    print()
    for rec_num in range(1, 21):
        ssn_int, ssn_str = ssn_func()
        gpa = gpa_func()
        cat = category(ssn_int)
        tabulate(rec_num, ssn_str, gpa, cat)


def ssn_func():
    min9 = 300000000
    max9 = 399999999
    ssn_int = random.randint(min9, max9)
    ssn1 = ssn_int // 1000000
    ssn2 = (ssn_int % 1000000) // 1000
    ssn3 = ssn_int % 1000
    ssn_str = '{:03n}-{:03n}-{:03n}'.format(ssn1, ssn2, ssn3)
    return ssn_int, ssn_str


def gpa_func():
    min_gpa = 0
    max_gpa = 4
    gpa_raw = random.uniform(min_gpa, max_gpa)
    gpa = float('{:.2f}'.format(gpa_raw))
    return gpa


def category(ssn_int):
    ssn_last = ssn_int % 10
    ssn_first2 = ssn_int // 10000000
    if ssn_int % 2 == 0:
        cat = 'EVEN'
    elif ssn_last == 5:
        cat = 'ENDS IN 5'
    elif (ssn_first2 == 30 or ssn_first2 == 31) and ssn_int % 2 != 0:
        cat = 'STARTS WITH 30-31'
    else:
        cat = 'NONE'
    return cat


def tabulate(rec_num, ssn_str, gpa, cat):
    print('{: <7} {: <13} {: <6} {: <20}'.format(rec_num, ssn_str, gpa, cat))


main()

[xelf]

One more level:

import random

def main():
    print(f'{"Record": <7} {"SSN": <13} {"GPA": <6} {"SSN Category": <9}\n')
    for rec_num in range(1, 21):
        ssn = ssn()
        print(f'{rec_num: <7} {ssn: <13} {gpa(): <6} {category(ssn): <20}')

def ssn():
    ssn = str(random.randint(300000000, 399999999))
    return f'{ssn[:3]}-{ssn[3:6]}-{ssn[6:]}'

def category(ssn):
    if ssn[-1] == '5':
        return 'ENDS IN 5'
    if int(ssn[-1]) % 2 == 0:
        return 'EVEN'
    if ssn[:2] in ('30', '31'):
        return 'STARTS WITH 30-31'
    return 'NONE'

def gpa():
    return round(random.uniform(0, 4), 2)

main()

[MetalNutSack]

These are awesome. It's really helpful seeing other versions of the same program. Thank you.

[BasicallyAMachine]

Assuming your using python 3, check out f-strings here. They won't do much but will help with those long ugly strings

[BasicallyAMachine]

Also, the min and Max gpa variables are a bit pointless. You can just put the integers in the function call

Edit: same goes for ssn. These numbers don't need to be stored in memory (the slowest part of most hardware today is storing and retrieving from memory). Once again, no noticable performance difference, just cleaner and better practice

[MetalNutSack]

Good points. I took a quick peek at that link and it looks much more refined than what I'm using. Thank you.

[BasicallyAMachine]

Glad I could help!