from db.orm import DrillingRealtimeORM
from db.pool import create_taos_pool


class TDengineWriter:
    def __init__(self, config, pool=None, orm=None):
        self.config = config
        self.pool = pool if pool is not None else create_taos_pool(config)
        self.orm = orm if orm is not None else DrillingRealtimeORM(
            config.database,
            stable=config.stable,
            default_device_code=config.device_code,
        )
        self.enabled = bool(config.enabled and self.pool)

    def write_payload(self, payload):
        if not self.enabled:
            return None
        sql = self.orm.build_insert_sql(payload)
        return self.pool.execute(sql)