import sqlite3
import os
from typing import Optional


class HansData():
    def __init__(self, hansNumber: str, proName: str, proComd: str, id: int = None) -> None:
        self.hansNumber = hansNumber
        self.proName = proName
        self.proComd = proComd
        self.id = id
        pass


class HansDB():
    def __init__(self) -> None:
        self.createDB()
        self.initDB()
        pass

    def createDB(self):
        global cur
        self.con = sqlite3.connect("HansLauncher.db")
        self.cur = self.con.cursor()

    def initDB(self):
        # SystemExit("Dont Exists HansLauncher.db")
        # sql = "create table hans(id integer primary key autoincrement,hans_number text,pro_name text,pro_comd text)"
        # self.cur.execute(sql)
        # self.con.commit()
        # self.cur.execute("show tables;")
        # print(self.cur.fetchall(), len(self.cur.fetchall()))
        pass

    def addHans(self, hans: HansData) -> bool:
        if self.getHansByNumber(hans.hansNumber) !=  None:
            return False
        else:
            sql = "insert into hans(hans_number,pro_name,pro_comd) values(?,?,?)"
            self.cur.execute(
                sql, (hans.hansNumber, hans.proName, hans.proComd))
            self.con.commit()
            return True

    def delHans(self, id: int):
        sql = "delete from hans where id = ?"
        self.cur.execute(sql, (str(id),))
        self.con.commit()
        print(self.cur.fetchall())

    def getHans(self) -> list[HansData]:
        sql = "select * from hans"
        self.cur.execute(sql)
        arr = []
        for i in self.cur.fetchall():
            arr.append(HansData(i[1], i[2], i[3], i[0]))
        return arr

    def getHansByNumber(self, hansNumber: str) -> Optional[HansData]:
        sql = "select * from hans where hans_number = ?"
        self.cur.execute(sql, (hansNumber,))
        arr = self.cur.fetchall()
        if len(arr) > 0:
            return HansData(arr[0][1], arr[0][2], arr[0][3], arr[0][0])