deprem-ocr / db_utils.py
mertcobanov's picture
UI/UX adjustment for data processing part (#10)
23690be
from deta import Deta # Import Deta
from pprint import pprint
import os
deta_key = os.getenv("DETA_KEY")
deta = Deta(deta_key)
db = deta.Base("deprem-ocr")
def get_users_by_city(city_name, limit=10):
user = db.fetch({"city": city_name.capitalize()}, limit=limit).items
return user
def get_all():
res = db.fetch()
all_items = res.items
# fetch until last is 'None'
while res.last:
res = db.fetch(last=res.last)
all_items += res.items
return all_items
def write_db(data_dict):
# 2) initialize with a project key
deta_key = os.getenv("DETA_KEY")
deta = Deta(deta_key)
# 3) create and use as many DBs as you want!
users = deta.Base("deprem-ocr")
users.insert(data_dict)
print("Pushed to db")
def get_latest_row(last):
all_items = get_all()
latest_items = all_items[-last:]
return latest_items