Python quickly generate codes

import random
import string

generate_code DEF (Bit_Num):
    '' '
    : param Bit_Num: generating a verification code digits
    : return: return codes generated
    ' ''

    all_str = string.printable.split('!')[0]
    code = ''.join([random.choice(all_str) for i in range(bit_num)])

    return code

if __name__ == '__main__':
    code = generate_code(6)
    print(code)

Generating codes with special symbols are as follows

import random
import string

generate_code DEF (Bit_Num):
    '' '
    : param Bit_Num: generating a verification code digits
    : return: return codes generated
    ' ''

    all_str = string.printable
    code = ''.join([random.choice(all_str) for i in range(bit_num)])

    return code

if __name__ == '__main__':
    code = generate_code(6)
    print(code)