Simple code generator function in Python. Optionally you can add custom string at the beginning.

code_generator.py

import uuid # from http://zesty.ca/python/uuid.html
import sys
import base64 


def fetch_code(custom_string="CODE_'):
    """
    takes an optional argument, a string to be put at the beginning of a code,
    so the code will look like that:

    CODE_8mANDJ
    KEY_9coOeC

    default is CODE_

    usage:
    fetch_code()   
    fetch_code(custom_string="KEY_")
  
    """
    b64uid = '00000000'
    
   
    uid = uuid.uuid4()
    b64uid = base64.b64encode(uid.bytes,'-_')
    
    code = b64uid[0:6]
    return custom_string+code