540 lines
17 KiB
Python
540 lines
17 KiB
Python
import json, os, shutil
|
|
import curses
|
|
|
|
os.system("title " + "GTA BuyBase")
|
|
|
|
logo = [
|
|
"|---------------------------------------------------------------------|",
|
|
"| KOTIKOT, script by BarsTiger |",
|
|
"| |",
|
|
"| |",
|
|
"| _____ _______ ____ ____ |",
|
|
"| / ____|__ __|/\ | _ \ | _ \ |",
|
|
"| | | __ | | / \ | |_) |_ _ _ _ | |_) | __ _ ___ ___ |",
|
|
"| | | |_ | | | / /\ \ | _ <| | | | | | | | _ < / _` / __|/ _ \ |",
|
|
"| | |__| | | |/ ____ \ | |_) | |_| | |_| | | |_) | (_| \__ \ __/ |",
|
|
"| \_____| |_/_/ \_\ |____/ \__,_|\__, | |____/ \__,_|___/\___| |",
|
|
"| __/ | |",
|
|
"| |___/ |",
|
|
"|---------------------------------------------------------------------|"
|
|
]
|
|
|
|
def glstdscr(stdscr):
|
|
return stdscr
|
|
|
|
globalstdscr = curses.wrapper(glstdscr)
|
|
|
|
def print_center(stdscr=globalstdscr, text=""):
|
|
stdscr.clear()
|
|
h, w = stdscr.getmaxyx()
|
|
x = w//2 - len(text)//2
|
|
y = h//2
|
|
stdscr.addstr(y, x, text)
|
|
stdscr.refresh()
|
|
|
|
def printlogo():
|
|
print("\n" * ((os.get_terminal_size().lines // 2 - (len(logo) // 2)) + 1))
|
|
for line in logo:
|
|
print(" " * ((os.get_terminal_size().columns//2 - (len(line)//2)) - 1) + line)
|
|
print("\n" * ((os.get_terminal_size().lines // 2 - (len(logo) // 2)) - 1))
|
|
input()
|
|
cls()
|
|
|
|
def cls():
|
|
os.system('cls' if os.name == 'nt' else 'clear')
|
|
|
|
def softcls():
|
|
print("\n" * (os.get_terminal_size().lines * 2))
|
|
|
|
def print_menu(stdscr, selected_row_idx, menu):
|
|
stdscr.clear()
|
|
h, w = stdscr.getmaxyx()
|
|
for idx, row in enumerate(menu):
|
|
x = w//2 - len(row)//2
|
|
y = h//2 - len(menu)//2 + idx
|
|
if idx == selected_row_idx:
|
|
stdscr.attron(curses.color_pair(1))
|
|
stdscr.addstr(y, x, row)
|
|
stdscr.attroff(curses.color_pair(1))
|
|
else:
|
|
stdscr.addstr(y, x, row)
|
|
stdscr.refresh()
|
|
|
|
doing = 0
|
|
menulist = {"main": ['Data output', 'Database loading and options', 'Add new item', 'Owning options', 'Exit'],
|
|
"output": ['Print all database', 'Print all items', 'Print all items by type', 'Print all items by shop', 'Print all items by price below this', 'Back'],
|
|
"baseoptions": ['Create backup of opened database', 'Open another database', 'Create new database', 'Validate database', 'Back'],
|
|
"ownoptions": ['Edit own or not', 'Show only owned', 'Show only unowned', 'Show all', 'Back'],
|
|
"exit": ["Exit", "Back"]}
|
|
|
|
def exitmenu(stdscr):
|
|
curses.curs_set(0)
|
|
curses.init_pair(1, curses.COLOR_BLACK, curses.COLOR_WHITE)
|
|
current_row = 0
|
|
|
|
print_menu(stdscr, current_row, menulist["exit"])
|
|
|
|
while 1:
|
|
key = stdscr.getch()
|
|
|
|
if key == curses.KEY_UP and current_row > 0:
|
|
current_row -= 1
|
|
elif key == curses.KEY_DOWN and current_row < len(menulist["exit"])-1:
|
|
current_row += 1
|
|
|
|
elif key == curses.KEY_ENTER or key in [10, 13]:
|
|
if current_row == 0:
|
|
exit()
|
|
if current_row == 1:
|
|
mainmenu(stdscr)
|
|
break
|
|
|
|
print_menu(stdscr, current_row, menulist["exit"])
|
|
|
|
def mainmenu(stdscr):
|
|
global doing
|
|
curses.curs_set(0)
|
|
curses.init_pair(1, curses.COLOR_BLACK, curses.COLOR_WHITE)
|
|
current_row = 0
|
|
|
|
print_menu(stdscr, current_row, menulist["main"])
|
|
|
|
while True:
|
|
key = stdscr.getch()
|
|
|
|
if key == curses.KEY_UP and current_row > 0:
|
|
current_row -= 1
|
|
elif key == curses.KEY_DOWN and current_row < len(menulist["main"])-1:
|
|
current_row += 1
|
|
|
|
elif key == curses.KEY_ENTER or key in [10, 13]:
|
|
if current_row == 0:
|
|
outputmenu(stdscr)
|
|
break
|
|
|
|
if current_row == 1:
|
|
baseoptionsmenu(stdscr)
|
|
break
|
|
|
|
if menulist["main"][current_row] == "Add new item":
|
|
doing = menulist["main"][current_row]
|
|
break
|
|
|
|
if current_row == 3:
|
|
ownmenu(stdscr)
|
|
break
|
|
|
|
if current_row == len(menulist["main"])-1:
|
|
exitmenu(stdscr)
|
|
break
|
|
|
|
print_menu(stdscr, current_row, menulist["main"])
|
|
|
|
def outputmenu(stdscr):
|
|
global doing
|
|
curses.curs_set(0)
|
|
curses.init_pair(1, curses.COLOR_BLACK, curses.COLOR_WHITE)
|
|
current_row = 0
|
|
|
|
print_menu(stdscr, current_row, menulist["output"])
|
|
|
|
while True:
|
|
key = stdscr.getch()
|
|
|
|
if key == curses.KEY_UP and current_row > 0:
|
|
current_row -= 1
|
|
elif key == curses.KEY_DOWN and current_row < len(menulist["output"])-1:
|
|
current_row += 1
|
|
|
|
elif key == curses.KEY_ENTER or key in [10, 13]:
|
|
if current_row == len(menulist["output"])-1:
|
|
mainmenu(stdscr)
|
|
break
|
|
else:
|
|
doing = menulist["output"][current_row]
|
|
break
|
|
|
|
elif key == curses.KEY_BACKSPACE or str(key) in ['KEY_BACKSPACE', '8', '127']:
|
|
mainmenu(stdscr)
|
|
break
|
|
|
|
print_menu(stdscr, current_row, menulist["output"])
|
|
|
|
def baseoptionsmenu(stdscr):
|
|
global doing
|
|
curses.curs_set(0)
|
|
curses.init_pair(1, curses.COLOR_BLACK, curses.COLOR_WHITE)
|
|
current_row = 0
|
|
|
|
print_menu(stdscr, current_row, menulist["baseoptions"])
|
|
|
|
while True:
|
|
key = stdscr.getch()
|
|
|
|
if key == curses.KEY_UP and current_row > 0:
|
|
current_row -= 1
|
|
elif key == curses.KEY_DOWN and current_row < len(menulist["baseoptions"])-1:
|
|
current_row += 1
|
|
|
|
elif key == curses.KEY_ENTER or key in [10, 13]:
|
|
if current_row == len(menulist["baseoptions"])-1:
|
|
mainmenu(stdscr)
|
|
break
|
|
else:
|
|
doing = menulist["baseoptions"][current_row]
|
|
break
|
|
|
|
elif key == curses.KEY_BACKSPACE or str(key) in ['KEY_BACKSPACE', '8', '127']:
|
|
mainmenu(stdscr)
|
|
break
|
|
|
|
print_menu(stdscr, current_row, menulist["baseoptions"])
|
|
|
|
def ownmenu(stdscr):
|
|
global doing
|
|
curses.curs_set(0)
|
|
curses.init_pair(1, curses.COLOR_BLACK, curses.COLOR_WHITE)
|
|
current_row = 0
|
|
|
|
print_menu(stdscr, current_row, menulist["ownoptions"])
|
|
|
|
while True:
|
|
key = stdscr.getch()
|
|
|
|
if key == curses.KEY_UP and current_row > 0:
|
|
current_row -= 1
|
|
elif key == curses.KEY_DOWN and current_row < len(menulist["ownoptions"])-1:
|
|
current_row += 1
|
|
|
|
elif key == curses.KEY_ENTER or key in [10, 13]:
|
|
if current_row == len(menulist["ownoptions"])-1:
|
|
mainmenu(stdscr)
|
|
break
|
|
else:
|
|
doing = menulist["ownoptions"][current_row]
|
|
break
|
|
|
|
elif key == curses.KEY_BACKSPACE or str(key) in ['KEY_BACKSPACE', '8', '127']:
|
|
mainmenu(stdscr)
|
|
break
|
|
|
|
print_menu(stdscr, current_row, menulist["ownoptions"])
|
|
|
|
def getkey(value, dictionary):
|
|
for key, val in dictionary.items():
|
|
if value == val:
|
|
return key
|
|
|
|
def getkeys(value, dictionary):
|
|
listofkeys = []
|
|
for key, val in dictionary.items():
|
|
if value == val:
|
|
listofkeys.append(key)
|
|
return listofkeys
|
|
|
|
def dictsortedkey(dictionary):
|
|
sorteddict = {}
|
|
for key in sorted(dictionary.keys()):
|
|
sorteddict[key] = dictionary[key]
|
|
return sorteddict
|
|
|
|
def printbase(dictionary):
|
|
keys = list(dictionary)
|
|
if showown == "All":
|
|
for key in keys:
|
|
print(key + ": " + dictionary[key][0] + " " + dictionary[key][1] + " " + "{:,}".format(dictionary[key][2]))
|
|
print()
|
|
|
|
def checkbelow(belowthis, dictionary):
|
|
yesbelow = {}
|
|
for key in list(dictionary):
|
|
if dictionary[key] <= int(belowthis):
|
|
yesbelow[key] = dictionary[key]
|
|
return yesbelow
|
|
|
|
basename = 'default.database'
|
|
backname = str(os.path.splitext(basename)[0]) + ".databack"
|
|
showown = "All"
|
|
|
|
if not os.path.isfile(basename):
|
|
basewrite = open(basename, 'w+')
|
|
empty = {}
|
|
json.dump(empty, basewrite, indent=3)
|
|
basewrite.close()
|
|
|
|
baseread = open(basename)
|
|
database = json.load(baseread)
|
|
baseread.close()
|
|
|
|
database = dictsortedkey(database)
|
|
|
|
buyitems = list(database)
|
|
typesofitems = []
|
|
shops = []
|
|
bytype = {}
|
|
byshop = {}
|
|
byprice = {}
|
|
|
|
for item in buyitems:
|
|
bytype[item] = database[item][0]
|
|
|
|
for item in buyitems:
|
|
byshop[item] = database[item][1]
|
|
|
|
for item in buyitems:
|
|
byprice[item] = database[item][2]
|
|
|
|
for key in buyitems:
|
|
typesofitems.append(bytype[key])
|
|
typesofitems = set(typesofitems)
|
|
typesofitems = list(typesofitems)
|
|
typesofitems = sorted(typesofitems)
|
|
|
|
for key in buyitems:
|
|
shops.append(byshop[key])
|
|
shops = set(shops)
|
|
shops = list(shops)
|
|
shops = sorted(shops)
|
|
|
|
def openbase():
|
|
global baseread, database, buyitems, typesofitems, shops, bytype, byshop, byprice, backname
|
|
if not os.path.isfile(basename):
|
|
basewrite = open(basename, 'w+')
|
|
empty = {}
|
|
json.dump(empty, basewrite, indent=3)
|
|
basewrite.close()
|
|
|
|
baseread = open(basename)
|
|
database = json.load(baseread)
|
|
baseread.close()
|
|
|
|
database = dictsortedkey(database)
|
|
|
|
buyitems = list(database)
|
|
typesofitems = []
|
|
shops = []
|
|
bytype = {}
|
|
byshop = {}
|
|
byprice = {}
|
|
|
|
for item in buyitems:
|
|
bytype[item] = database[item][0]
|
|
|
|
for item in buyitems:
|
|
byshop[item] = database[item][1]
|
|
|
|
for item in buyitems:
|
|
byprice[item] = database[item][2]
|
|
|
|
for key in buyitems:
|
|
typesofitems.append(bytype[key])
|
|
typesofitems = set(typesofitems)
|
|
typesofitems = list(typesofitems)
|
|
typesofitems = sorted(typesofitems)
|
|
|
|
for key in buyitems:
|
|
shops.append(byshop[key])
|
|
shops = set(shops)
|
|
shops = list(shops)
|
|
shops = sorted(shops)
|
|
backname = str(os.path.splitext(basename)[0]) + ".databack"
|
|
|
|
printlogo()
|
|
|
|
while True:
|
|
curses.wrapper(mainmenu)
|
|
|
|
if doing == menulist["output"][0]:
|
|
print("Name: Type Shop Price")
|
|
print()
|
|
printbase(database)
|
|
print()
|
|
input("To go back to menu press Enter...")
|
|
softcls()
|
|
|
|
elif doing == menulist["output"][1]:
|
|
for item in buyitems:
|
|
print(item)
|
|
print()
|
|
input("To go back to menu press Enter...")
|
|
softcls()
|
|
|
|
elif doing == menulist["output"][2]:
|
|
print("Types to search in database: ")
|
|
for item in typesofitems:
|
|
print(item)
|
|
print()
|
|
for item in getkeys(input("For which type I should check: "), bytype):
|
|
print(item)
|
|
print()
|
|
input("To go back to menu press Enter...")
|
|
softcls()
|
|
|
|
elif doing == menulist["output"][3]:
|
|
print("Types to search in database: ")
|
|
for item in shops:
|
|
print(item)
|
|
print()
|
|
for item in getkeys(input("For which shop I should check: "), byshop):
|
|
print(item)
|
|
print()
|
|
input("To go back to menu press Enter...")
|
|
softcls()
|
|
|
|
elif doing == menulist["output"][4]:
|
|
belowthisprice = checkbelow(input("I will check for items below this price: "), byprice)
|
|
for item in list(belowthisprice):
|
|
print(item + ": " + "{:,}".format(belowthisprice[item]))
|
|
print()
|
|
input("To go back to menu press Enter...")
|
|
softcls()
|
|
|
|
elif doing == "Add new item":
|
|
print()
|
|
print("Adding new item")
|
|
specif = []
|
|
nameofnewitem = input("Type here name of new item: ")
|
|
specif.append(input("Type here type of item: "))
|
|
specif.append(input("Type here shop, where item can be bought: "))
|
|
specif.append(int(input("Type here price of item (without spaces or comas): ")))
|
|
isowneditem = bool(input("Do you own this item? (1 - Yes, 0 - No): ") == "1")
|
|
specif.append(isowneditem)
|
|
database[nameofnewitem] = specif
|
|
basewrite = open(basename, 'w+')
|
|
json.dump(database, basewrite, indent=3, ensure_ascii=False)
|
|
basewrite.close()
|
|
|
|
elif doing == menulist["baseoptions"][0]:
|
|
print()
|
|
shutil.copy(basename, backname)
|
|
if os.path.isfile(backname):
|
|
print("Backup successful")
|
|
print()
|
|
input("To go back to menu press Enter...")
|
|
softcls()
|
|
|
|
elif doing == menulist["baseoptions"][1]:
|
|
alldatabases = []
|
|
for file in os.listdir(os.getcwd()):
|
|
if '.database' in file:
|
|
alldatabases.append(os.path.splitext(file)[0])
|
|
print("Which database would you like to open?")
|
|
for base in alldatabases:
|
|
print(base)
|
|
basename = input("Type here: ") + ".database"
|
|
openbase()
|
|
print("Other DB opened")
|
|
input("To go back to menu press Enter...")
|
|
softcls()
|
|
|
|
elif doing == menulist["baseoptions"][2]:
|
|
print("Creating new DB. Choose name for it")
|
|
basename = input("Type here: ") + ".database"
|
|
openbase()
|
|
print("New DB created")
|
|
input("To go back to menu press Enter...")
|
|
softcls()
|
|
|
|
elif doing == menulist["baseoptions"][3]:
|
|
print("Checking file valid")
|
|
try:
|
|
openbase()
|
|
print("File is valid")
|
|
except:
|
|
print("File is !INvalid!")
|
|
print("Try to check file by yourself or create new DB")
|
|
|
|
print()
|
|
print("Checking if all owning options are valid")
|
|
|
|
for item in buyitems:
|
|
try:
|
|
if database[item][3] == True:
|
|
pass
|
|
elif database[item][3] == False:
|
|
pass
|
|
else:
|
|
database[item][3] = False
|
|
except:
|
|
databasethisitem = database[item]
|
|
databasethisitem.append(False)
|
|
database[item] = databasethisitem
|
|
|
|
basewrite = open(basename, 'w+')
|
|
json.dump(database, basewrite, indent=3, ensure_ascii=False)
|
|
basewrite.close()
|
|
|
|
print("Owning options were checked and validated if needed")
|
|
|
|
input("To go back to menu press Enter...")
|
|
softcls()
|
|
|
|
elif doing == menulist["ownoptions"][0]:
|
|
print("Select type of editor")
|
|
typeofedit = input("1 - Edit all items\n2 - Edit one at a time\nType here: ")
|
|
if typeofedit == "2":
|
|
print()
|
|
print("Which item you would like to change?")
|
|
whattodonow = input("1 - Print all items\n2 - Search item\nType here: ")
|
|
if whattodonow == "1":
|
|
for item in buyitems:
|
|
print(item)
|
|
print()
|
|
elif whattodonow == "2":
|
|
whichtocontain = input("Start typing name here: ")
|
|
containthis = []
|
|
for item in buyitems:
|
|
if whichtocontain.lower() in item.lower():
|
|
containthis.append(item)
|
|
for item in containthis:
|
|
print(item)
|
|
itemtoedit = input("Type here name of item: ")
|
|
print(itemtoedit + " is now " + str(database[itemtoedit][3]))
|
|
isowneditem = bool(input("Do you own this item? (1 - Yes, 0 - No): ") == "1")
|
|
database[itemtoedit][3] = isowneditem
|
|
|
|
basewrite = open(basename, 'w+')
|
|
json.dump(database, basewrite, indent=3, ensure_ascii=False)
|
|
basewrite.close()
|
|
|
|
elif typeofedit == "1":
|
|
ownbasename = str(os.path.splitext(basename)[0]) + ".ownbase"
|
|
print("Will be created file named " + ownbasename)
|
|
owningtempbase = {}
|
|
for item in buyitems:
|
|
owningtempbase[item] = database[item][3]
|
|
|
|
with open(ownbasename, 'w+') as ownwrite:
|
|
json.dump(owningtempbase, ownwrite, indent=3, ensure_ascii=False)
|
|
|
|
print()
|
|
print("File is created. Open it with any text editor")
|
|
print("If you own item, type true near its name. If not - false")
|
|
print("It shold be like this: ")
|
|
print('"Avenger": true,')
|
|
print("After each line shold be a ','")
|
|
print()
|
|
|
|
input("When you are ready, press Enter")
|
|
input("Press Enter to load data from ownbase file")
|
|
|
|
with open(ownbasename) as ownread:
|
|
owningtempbase = json.load(ownread)
|
|
|
|
for item in buyitems:
|
|
database[item][3] = owningtempbase[item]
|
|
|
|
basewrite = open(basename, 'w+')
|
|
json.dump(database, basewrite, indent=3, ensure_ascii=False)
|
|
basewrite.close()
|
|
|
|
os.remove(ownbasename)
|
|
|
|
print("All data from ownbase file moved to database file")
|
|
print("Ownbase file deleted")
|
|
print()
|
|
|
|
|
|
input("To go back to menu press Enter...")
|
|
softcls()
|