mirror of
https://github.com/OMGeeky/gpt-pilot.git
synced 2025-12-28 15:17:53 +01:00
30 lines
643 B
Python
30 lines
643 B
Python
import os
|
|
from peewee import *
|
|
from datetime import datetime
|
|
from uuid import uuid4
|
|
|
|
DB_NAME = os.getenv("DB_NAME")
|
|
DB_HOST = os.getenv("DB_HOST")
|
|
DB_PORT = os.getenv("DB_PORT")
|
|
DB_USER = os.getenv("DB_USER")
|
|
DB_PASSWORD = os.getenv("DB_PASSWORD")
|
|
|
|
|
|
# Establish connection to the database
|
|
database = PostgresqlDatabase(
|
|
DB_NAME,
|
|
user=DB_USER,
|
|
password=DB_PASSWORD,
|
|
host=DB_HOST,
|
|
port=DB_PORT
|
|
)
|
|
|
|
|
|
class BaseModel(Model):
|
|
id = UUIDField(primary_key=True, default=uuid4)
|
|
created_at = DateTimeField(default=datetime.now)
|
|
updated_at = DateTimeField(default=datetime.now)
|
|
|
|
class Meta:
|
|
database = database
|