import json
import os
class DB:
"""
Klassa dla pracy z tekstowa DB
"""
def __init__(self, path: str):
"""
Args:
path (str): Sciezka do Bazy Danych
"""
self.path = path
with open(self.path, 'r+') as file:
f = file.read().strip()
if len(f) != 0:
if f[0] != '{' or f[-1] != '}':
f = '{}'
else:
f = '{}'
self.data = json.loads(f)
def __str__(self):
return json.dumps(self.data)
def add_new_person(self):
"""
Dodajemy boksera
"""
name = input("Name: ")
surname = input("Surname: ")
weight = int(input("Weight: "))
age = int(input("Age: "))
wins = int(input("Wins: "))
power = int(input("Power: "))
NickName = input("NickName: ")
if NickName in self.data:
print("Osoba o takim NickName`ie juz istnieje. Sprawdz poprawnosc dannych.")
if NickName not in self.data:
self.data[NickName] = {'name': name, 'surname': surname, 'weight': weight, 'age': age, 'wins': wins, 'power': power}
self.save()
def remove_person(self):
"""
Usuniecie osoby z DB
"""
NickName = input("NickName: ")
if NickName not in self.data:
print("Bokser o takim NickName`ie nie istnieje, ewentualnie sprawdz poprawnosc wprowadzonych dannych")
else:
del self.data[NickName]
self.save()
def show_person(self):
"""
Wyswietlanie informacji o Bokserze po NickName
"""
NickName = input("NickName: ")
if NickName not in self.data:
print("Bokser o takim NickName`ie nie istnieje. Mozesz go dodac, albo ewentualnie sprawdz poprawnosc wprowadzonych dannych")
else:
# indent dla ladnego wyswietlania
print(json.dumps(self.data[NickName], indent=2))
def show_everybody(self):
"""
Wyswietlanie calej bazy danych
"""
if len(self.data) == 0:
print(
"Niema zadnego zapisu w bazie danych, ale mozesz dodac jakis!")
else:
# indent dla ladnego wyswietlania
print(json.dumps(self.data, indent=2))
def copy_file(self):
"""
Kopiowanie pliku (sprawdzamy czy istnieje)
"""
file1 = input("wpisz nazwe pliku (na przykład db.json):")
file2 = input("wpisz nazwe pliku (na przykład db_copy.json):")
if os.path.isfile(file1):
if os.path.isfile(file2):
with open(file1) as inp, open(file2, "w") as out:
out.write(inp.read())
else:
print("plik o nazwie", file2, "nie istnieje")
def show_user_file(self):
"""
Wyswietli dane z pliku podanego przez uzytkownika
"""
user_file = input("podaj nazwe pliku (.json), ktory chcesz wyswietlic (naprzyklad db_copy.json): ")
if os.path.isfile(user_file):
if os.stat(user_file).st_size == 0:
print("Ten plik jest pusty!")
else:
with open(user_file) as uf:
print(uf.read())
else:
print("plik o nazwie", user_file, "nie istnieje")
def save(self):
with open(self.path, 'w') as file:
# indent dla ladnego zapisu
json.dump(self.data, file, indent=2)
def fight(self):
print("Podaj NickName dwoch bokserow by zorganizowac walke.")
first_fighter = 'SpeedFire'
#second_fighter = json.loads(input("Wpisz NickName drugiego: "))
if first_fighter in self.data:
print(json.dumps(self.data[first_fighter], indent=2))
path = "db.json"
p = DB(path)
funcs = [p.add_new_person, p.remove_person, p.show_person, p.show_everybody, p.copy_file, p.show_user_file, p.fight]
while True:
choice = int(
input("wpisz działanie\n1 : dodaj | 2 : usun | 3 : pokaz | 4 : pokaz_wszystkich | 5 : kopiuj_plik | 6 : wyswietlij_z_pliku | 7 : walka | 0 : zamknij: "))
if not choice:
break
funcs[choice-1]()
