from peewee import * db = SqliteDatabase("checkinbot.db") db.connect(reuse_if_open=True) class BaseModel(Model): class Meta: database = db class User(BaseModel): telegram_user_id = CharField(unique=True, primary_key=True) access_key = CharField(max_length=64) home_instance = CharField(max_length=128) state = CharField(max_length=128) client_id = CharField(max_length=128) client_secret = CharField(max_length=128) class Location(BaseModel): fsq_id = CharField(unique=True, primary_key=True) name = CharField(max_length=128) locality = CharField(max_length=128) region = CharField(max_length=128) latitude = CharField(max_length=128) longitude = CharField(max_length=128) with db.connection_context(): db.create_tables([User, Location]) def get_poi_by_fsq_id(fsq_id) -> dict: with db.connection_context(): try: poi = Location.get(Location.fsq_id == fsq_id) return { "name": poi.name, "locality": poi.locality, "region": poi.region, "latitude": poi.latitude, "longitude": poi.longitude, } except DoesNotExist: return {} def create_poi(poi: dict): with db.connection_context(): poi = Location.create( fsq_id=poi["fsq_id"], name=poi["name"], locality=poi["locality"], region=poi["region"], latitude=poi["latitude"], longitude=poi["longitude"], ) poi.save()