ClaytonWWilson/Scraper-for-theTVDB.com
1
0
Fork 0
mirror of https://github.com/ClaytonWWilson/Scraper-for-theTVDB.com.git synced 2025-12-15 17:28:46 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Added some if statements

Browse Source
This commit is contained in:
Clayton Wilson 2017-11-01 08:19:29 -04:00
parent d559916c64
commit 375a34daba
1 changed files with 34 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

69
login.py
View File

@ -4,49 +4,48 @@ from checks import *
import time import time
def login(): def login():
if (os.path.exists("login.json") == False): if os.path.exists("login.json") == False:
obj = open("login.json") obj = open("login.json")
# obj.write() # obj.write()
obj.close() obj.close()
login = { if os.stat("login.json").st_size == 0:# Will only ask for credentials if the login file is empty
"API_KEY": "", login = {
"USER_KEY": "", "API_KEY": "",
"USER_NAME": "", "USER_KEY": "",
"TOKEN": "", "USER_NAME": "",
"TIMESTAMP": "" "TOKEN": "",
} "TIMESTAMP": ""
}
tmp_api_key = login["API_KEY"] tmp_api_key = login["API_KEY"]
tmp_user_key = login["USER_KEY"] tmp_user_key = login["USER_KEY"]
tmp_user_name = login["USER_NAME"] tmp_user_name = login["USER_NAME"]
while tmp_api_key is "": while tmp_api_key is "":
tmp_api_key = input("Please enter your api key: ") tmp_api_key = input("Please enter your api key: ")
while tmp_user_key is "": while tmp_user_key is "":
tmp_user_key = input("Please enter your user key: ") tmp_user_key = input("Please enter your user key: ")
while tmp_user_name is "": while tmp_user_name is "":
tmp_user_name = input("Please enter your username: ") tmp_user_name = input("Please enter your username: ")
LOGIN_DATA = { LOGIN_DATA = {
"apikey": tmp_api_key, "apikey": tmp_api_key,
"userkey": tmp_user_key, "userkey": tmp_user_key,
"username": tmp_user_name "username": tmp_user_name
} }
tmp_token = getToken(LOGIN_DATA) tmp_token = getToken(LOGIN_DATA)
if tmp_token is "": if tmp_token is "":
print("Authentication failed. Please try again.") print("Authentication failed. Please try again.")
else:
with open("login.json") as json_data: with open("login.json") as json_data:
login["API_KEY"] = tmp_api_key login["API_KEY"] = tmp_api_key
login["USER_KEY"] = tmp_user_key login["USER_KEY"] = tmp_user_key
login["USER_NAME"] = tmp_user_name login["USER_NAME"] = tmp_user_name
obj = open("login.json", "w") obj = open("login.json", "w")
obj.write(json.dumps(login)) obj.write(json.dumps(login))
obj.close() obj.close()
def getCredentials():
print()
# TODO try to get token, if token fails, ask for login info again, if it passes save login details to login.py and save token with timestamp. # TODO try to get token, if token fails, ask for login info again, if it passes save login details to login.py and save token with timestamp.
# TODO at startup, check token for validity and remove it if it is expired # TODO at startup, check token for validity and remove it if it is expired