import pandas as pd
import sqlalchemy

"""
Спарешенные пары:
USDT_BTC
USDT_LTC
USDT_ETH
USDT_ETC
USDT_XRP
"""


def connect(user, password, db, host: str, port: int, echo=True):
    url = 'postgresql+psycopg2://{}:{}@{}:{}/{}'
    url = url.format(user, password, host, port, db)
    eng = sqlalchemy.create_engine(url, client_encoding='utf8', echo=echo)
    meta = sqlalchemy.MetaData(bind=eng)

    return eng, meta


def get_data_frame(pair: str = 'USDT_BTC') -> pd.DataFrame:
    """Метод стягивания данных из базы в датафрейм.

    По умолчанию тянет все значения в базе для валютной пары доллар биткоин.
    Список спаршенных пар смотри в таблице Pair
    Цепляться будет отовсюду где есть инетрнет"""
    engine, meta = connect(user='postgres', password='password', db='btccandles', host='176.212.125.30', port=16432)
    df = pd.read_sql_query(
        'select date, time, open, close, low, high, volume, pair."name" from candlestick, pair '
        'where candlestick.pair_id=pair.id and pair."name"=\'' + pair + '\';',
        con=engine)
    return df

# Демонстрируем парсинг!
# print(get_data_frame())