from sqlalchemy import create_engine, Table, Column, Integer, String, Float, Double, DateTime
from sqlalchemy.orm import registry, sessionmaker

# 메타데이터 객체 관리
mapper_registry = registry()


candle_table = Table(
    'candle',
    mapper_registry.metadata,
    Column("contract", String(20), primary_key=True),
    Column("open", String(50)),
    Column("close", String(50)),
    Column("high", String(50)),
    Column("low", String(50)),
    Column("volume", Integer),
    Column("amount", String(50)),
    Column("time", Integer, primary_key=True),

)

# Python 유저 객체
class Candle:
    contract: str
    open:str
    close:str
    high:str
    low:str
    volume:int
    amount:str
    time:int

# 테이블과 클래스의 매핑 설정
mapper_registry.map_imperatively(Candle, candle_table)

def get_session():
    # 데이터베이스 설정
    engine = get_engine()
    mapper_registry.metadata.create_all(engine)  # 테이블 생성
    Session = sessionmaker(bind=engine)
    session = Session()
    return session

def get_engine() :
    return create_engine('postgresql+psycopg2://bangae1:fpdlwms1@hmsn.ink:35432/coin')