Hell13Cat
36 lines
1018 B
Python
36 lines
1018 B
Python
from tinydb import TinyDB, Query
|
|
import os
|
|
import json
|
|
|
|
class DB:
|
|
def __init__(self):
|
|
db_path = "datas"
|
|
if not os.path.exists(db_path):
|
|
os.makedirs(os.path.dirname(db_path), exist_ok=True)
|
|
json.dump({}, open(db_path+"/db.json", "w", encoding="utf-8"))
|
|
self.db = TinyDB(db_path+"/db.json")
|
|
self.File = Query()
|
|
|
|
def add(self, data_add):
|
|
self.db.insert(data_add)
|
|
|
|
def add_raw(self, data_add):
|
|
# self.db.insert(data_add)
|
|
pass
|
|
|
|
def get_all(self):
|
|
return self.db.all()
|
|
|
|
def get_search(self, ratings, tags):
|
|
results = self.db.search(
|
|
self.File.rating.one_of(self.ratings) &
|
|
self.File.tags.any(self.tags)
|
|
)
|
|
return results
|
|
|
|
def get_unknow(self):
|
|
results = self.db.search(self.File.tags == [])
|
|
return results
|
|
|
|
def edit_data(self, id_file, edit_data):
|
|
self.db.update(self.edit_data, self.File.md5 == self.id_file) |