Douyin_TikTok_Download_API/config.py

import configparser

config = configparser.ConfigParser()
config_path = 'config.ini'
config.read(config_path, encoding='utf-8')


def api_config():
    api_default_port = config.get('Web_API', 'Port')
    api_new_port = input(f'Default API port: {api_default_port}\nIf you want use different port input new API port here: ')
    if api_new_port.isdigit():
        if int(api_new_port) == int(api_default_port):
            print(f'Use default port for web_app.py: {api_default_port}')
        else:
            print(f'Use new port for web_api.py: {api_new_port}')
            config.set('Web_API', 'Port', api_new_port)
            config.write(open(config_path, "w", encoding="utf-8"))
    else:
        print(f'Use default port for web_app.py: {api_default_port}')

    req_limit = config.get('Web_API', 'Rate_Limit')
    new_req_limit = input(f'Default API rate limit: {req_limit}\nIf you want use different rate limit input new rate limit here: ')
    if new_req_limit.isdigit():
        if int(new_req_limit) == int(req_limit.split('/')[0]):
            print(f'Use default rate limit for web_api.py : {req_limit}')
        else:
            print(f'Use new rate limit: {new_req_limit}/minute')
            config.set('Web_API', 'Rate_Limit', f'{new_req_limit}/minute')
            config.write(open(config_path, "w", encoding="utf-8"))
    else:
        print(f'Use default rate limit for web_api.py: {req_limit}')


def app_config():
    app_default_port = config.get('Web_APP', 'Port')
    app_new_port = input(f'Default App port: {app_default_port}\nIf you want use different port input new App port here: ')
    if app_new_port.isdigit():
        if int(app_new_port) == int(app_default_port):
            print(f'Use default port for web_app.py: {app_default_port}')
        else:
            print(f'Use new port: {app_new_port}')
            config.set('Web_APP', 'Port', app_new_port)
            config.write(open(config_path, "w", encoding="utf-8"))
    else:
        print(f'Use default port for web_app.py : {app_default_port}')


if __name__ == '__main__':
    api_config()
    app_config()