29 lines
595 B
Python
29 lines
595 B
Python
from datetime import datetime
|
|
from typing import Union
|
|
|
|
from peewee import Proxy, Database, Model, CharField, DateTimeField, SqliteDatabase, CompositeKey
|
|
|
|
db: Union[Database, Proxy] = Proxy()
|
|
|
|
|
|
class BaseModel(Model):
|
|
class Meta:
|
|
database = db
|
|
|
|
|
|
class Rides(BaseModel):
|
|
class Meta:
|
|
primary_key = CompositeKey('name', 'date')
|
|
name: str = CharField()
|
|
date: datetime = DateTimeField(default=datetime.now)
|
|
|
|
|
|
models = [Rides]
|
|
|
|
|
|
def connect(db_name: str):
|
|
sqlite = SqliteDatabase(db_name)
|
|
sqlite.connect()
|
|
db.initialize(sqlite)
|
|
db.create_tables(models)
|