diff --git a/func.py b/func.py new file mode 100644 index 0000000..d56913f --- /dev/null +++ b/func.py @@ -0,0 +1,73 @@ +#!/usr/bin/python3 +# -- coding: utf-8 -- + +import db, os, subprocess +from db import * + +def u_auth(id,passwd): + list = [] + if db.check('obj',AUTH_FILE): + list = db.read(AUTH_FILE) + if id in list: + return 'You already logged in' + else: + if passwd == os.environ['PASS']: + list.append(id) + db.write(list,AUTH_FILE) + return 'Success' + else: + return 'Wrong password' + +def auth_check(id): + if db.check('obj',AUTH_FILE): + list = db.read(AUTH_FILE) + else: + list = [] + if id in list: + return True + +def add_dir(id,dir,path): + if auth_check(id): + if os.path.exists(path) == False: + return 'Folder not exist on host' + if db.check('obj',DIR_FILE): + dict = db.read(DIR_FILE) + else: + dict = {} + dict.setdefault(dir,path) + db.write(dict,DIR_FILE) + return 'Success' + else: + return 'Log in first' + +def del_dir(id,dir): + if auth_check(id): + if db.check('obj',DIR_FILE): + dict = db.read(DIR_FILE) + else: + dict = {} + if dir in dict: + del dict[dir] + db.write(dict,DIR_FILE) + return 'Success' + else: + return 'Dir not exists' + else: + return 'Log in first' + +def magnet(id,link,dir): + if auth_check(id): + subprocess.run(f"qbt torrent add url {link} -f {dir}", check=True, shell=True) + return 'Success' + +def file(id,file,dir): + if auth_check(id): + subprocess.run(f"qbt torrent add file {PATH}{file} -f {dir}", check=True, shell=True) + os.remove(f'{PATH}{file}') + return 'Success' + +def dirlist(): + dirs = {} + if db.check('obj',DIR_FILE): + dirs = db.read(DIR_FILE) + return dirs