48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
import json
|
|
import modules.vars as horsy_vars
|
|
|
|
|
|
def get_auth(is_gui=False, login_ui=None):
|
|
with open(horsy_vars.horsypath + 'config.cfg') as f:
|
|
config = json.load(f)
|
|
|
|
try:
|
|
if config['auth']:
|
|
return config['auth']
|
|
else:
|
|
raise Exception('No auth found')
|
|
except:
|
|
if not is_gui:
|
|
print('[!] No auth found, please login first')
|
|
print('email')
|
|
email = input('> ')
|
|
print('password')
|
|
password = input('> ')
|
|
config['auth'] = {'email': email, 'password': password}
|
|
with open(horsy_vars.horsypath + 'config.cfg', 'w') as f:
|
|
json.dump(config, f)
|
|
print('[OK] Auth created')
|
|
return config['auth']
|
|
else:
|
|
login_ui.show()
|
|
login_ui.login_button.clicked.connect(lambda: get_auth(login_ui=login_ui))
|
|
|
|
|
|
def get_gui_auth(login_ui):
|
|
with open(horsy_vars.horsypath + 'config.cfg') as f:
|
|
config = json.load(f)
|
|
if login_ui.email_box.text() != '' and login_ui.password_box.text() != '':
|
|
with open(horsy_vars.horsypath + 'config.cfg', 'w') as f:
|
|
config['auth'] = {'email': login_ui.email_box.text(), 'password': login_ui.password_box.text()}
|
|
json.dump(config, f)
|
|
|
|
|
|
def del_auth():
|
|
with open(horsy_vars.horsypath + 'config.cfg') as f:
|
|
config = json.load(f)
|
|
if config['auth']:
|
|
config['auth'] = None
|
|
with open(horsy_vars.horsypath + 'config.cfg', 'w') as f:
|
|
json.dump(config, f)
|
|
print('[OK] Auth deleted')
|