from . import task_type
from .database import cursor, conn


class Ship:
    def __init__(self, symbol: str):
        self.symbol = symbol
        self.callsign = "-".split(symbol)[0]

        self.task = None

    def load_task(self):
        cursor.execute("SELECT task, params FROM ships WHERE symbol = ?", (self.symbol,))
        row = cursor.fetchone()
        if row is None:
            cursor.execute("INSERT INTO ships (symbol, task, params) VALUES (?, ?, ?)", (self.symbol, task_type.IDLE, None))
            conn.commit()

            self.task = task_type.IDLE
        else:
            self.task = row[0]

    def set_task(self, task):
        self.task = task
        if task != task_type.ERROR:
            cursor.execute("UPDATE ships SET task = ? WHERE symbol = ?", (task, self.symbol))
            conn.commit()