update.py

import sys, os
import shutil
# https://stackoverflow.com/questions/41836988/git-push-via-gitpython
for root, dirs, files in os.walk(".", topdown=False):
    for name in dirs:
        if name == '__pycache__':
            shutil.rmtree(os.path.join(root, name))

rmtrees = ["products", "data", "static/assets/prodimages","configs"]
for i in rmtrees:
    try:
        shutil.rmtree(i)
        if i == 'logs':
            os.mkdir('logs')
    except Exception as e:
        input(e)
    try:
        os.remove('setup')
    except Exception as e:
        input(e)

a = input("LoonaBilling Update Script \n1. Main Branch \n2. Development Branch\n>>> """)
if isinstance(a, int) == False:
    raise TypeError
if a == '1':
    os.system('git checkout main')
elif a == '2':
    os.system('git checkout development')
else:
    print('Invalid option')
    exit()

os.system('pipreqs . --force')
os.system('git add -A')
os.system('git commit -a')
input('Push enter')
if a == '1':
    os.system('git push origin HEAD:main')
elif a == '2':
    os.system('git push origin HEAD:development')
else:
    print('Invalid option')
    exit()