ad9b6a3751
Added navigation like in GTA5 menus: After pressing Up button on first raw you will be on last And after pressing Down on last you will be in first
617 lines
20 KiB
Python
617 lines
20 KiB
Python
import json, os, shutil, subprocess
|
|
import curses
|
|
import sys
|
|
|
|
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_UP and current_row == 0:
|
|
current_row = len(menulist["main"])-1
|
|
elif key == curses.KEY_DOWN and current_row < len(menulist["main"])-1:
|
|
current_row += 1
|
|
elif key == curses.KEY_DOWN and current_row == len(menulist["main"]) - 1:
|
|
current_row = 0
|
|
|
|
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_UP and current_row == 0:
|
|
current_row = len(menulist["output"])-1
|
|
elif key == curses.KEY_DOWN and current_row < len(menulist["output"])-1:
|
|
current_row += 1
|
|
elif key == curses.KEY_DOWN and current_row == len(menulist["output"]) - 1:
|
|
current_row = 0
|
|
|
|
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_UP and current_row == 0:
|
|
current_row = len(menulist["baseoptions"])-1
|
|
elif key == curses.KEY_DOWN and current_row < len(menulist["baseoptions"])-1:
|
|
current_row += 1
|
|
elif key == curses.KEY_DOWN and current_row == len(menulist["baseoptions"]) - 1:
|
|
current_row = 0
|
|
|
|
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_UP and current_row == 0:
|
|
current_row = len(menulist["ownoptions"])-1
|
|
elif key == curses.KEY_DOWN and current_row < len(menulist["ownoptions"])-1:
|
|
current_row += 1
|
|
elif key == curses.KEY_DOWN and current_row == len(menulist["ownoptions"]) - 1:
|
|
current_row = 0
|
|
|
|
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 show == "All":
|
|
for key in keys:
|
|
print(key + ": " + dictionary[key][0] + " " + dictionary[key][1] + " " + "{:,}".format(dictionary[key][2]))
|
|
print()
|
|
elif show == "Owned":
|
|
for key in keys:
|
|
if dictionary[key][3] == True:
|
|
print(key + ": " + dictionary[key][0] + " " + dictionary[key][1] + " " + "{:,}".format(dictionary[key][2]))
|
|
print()
|
|
elif show == "Unowned":
|
|
for key in keys:
|
|
if dictionary[key][3] == False:
|
|
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"
|
|
show = "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]:
|
|
if show == "All":
|
|
for item in buyitems:
|
|
print(item)
|
|
elif show == "Owned":
|
|
for item in buyitems:
|
|
if database[item][3] == True:
|
|
print(item)
|
|
elif show == "Unowned":
|
|
for item in buyitems:
|
|
if database[item][3] == False:
|
|
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()
|
|
if show == "All":
|
|
for item in getkeys(input("For which type I should check: "), bytype):
|
|
print(item)
|
|
elif show == "Owned":
|
|
for item in getkeys(input("For which type I should check: "), bytype):
|
|
if database[item][3] == True:
|
|
print(item)
|
|
elif show == "Unowned":
|
|
for item in getkeys(input("For which type I should check: "), bytype):
|
|
if database[item][3] == False:
|
|
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()
|
|
if show == "All":
|
|
for item in getkeys(input("For which type I should check: "), byshop):
|
|
print(item)
|
|
elif show == "Owned":
|
|
for item in getkeys(input("For which type I should check: "), byshop):
|
|
if database[item][3] == True:
|
|
print(item)
|
|
elif show == "Unowned":
|
|
for item in getkeys(input("For which type I should check: "), byshop):
|
|
if database[item][3] == False:
|
|
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)
|
|
if show == "All":
|
|
for item in list(belowthisprice):
|
|
print(item + ": " + "{:,}".format(belowthisprice[item]))
|
|
elif show == "Owned":
|
|
for item in list(belowthisprice):
|
|
if database[item][3] == True:
|
|
print(item + ": " + "{:,}".format(belowthisprice[item]))
|
|
elif show == "Unowned":
|
|
for item in list(belowthisprice):
|
|
if database[item][3] == False:
|
|
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()
|
|
softcls()
|
|
|
|
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 (notepad will be opened automatically on Windows)")
|
|
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()
|
|
if sys.platform == "win32":
|
|
input("After you press Enter, notepad app will be opened")
|
|
command = 'cmd /c "notepad.exe {}"'.format(ownbasename)
|
|
os.system(command)
|
|
|
|
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()
|
|
|
|
elif doing == menulist["ownoptions"][1]:
|
|
show = "Owned"
|
|
elif doing == menulist["ownoptions"][2]:
|
|
show = "Unowned"
|
|
elif doing == menulist["ownoptions"][3]:
|
|
show = "All"
|