import sqlite3, os

class Database():
    def __init__(self):
        os.makedirs('resources', exist_ok=True)
        self.conn = sqlite3.connect('./resources/company.db')
        def dict_factory(cursor, row):#重定义row_factory函数查询返回数据类型是字典形式
            d = {}
            for idx, col in enumerate(cursor.description):
                d[col[0]] = row[idx]
            return d
        self.conn.row_factory = dict_factory

        self.c = self.conn.cursor()
        self.creatble()

    def creatble(self):
        self.c.execute('''CREATE TABLE IF NOT EXISTS student
       ( 
        id_number        CHAR(50)    NOT NULL ,
   
        user_name       CHAR(50)    NOT NULL,
        gender           bool,
        password        char(20)    NOT NULL,
        img_path        char(60),
        vector          blob        ,
           
        salt            char(10)  NOT NULL ,
        count   INT,
        PRIMARY KEY (id_number )
                 )without rowid;''')

        self.c.execute('''CREATE TABLE IF NOT EXISTS student_log_time 
       ( 
        id_number              CHAR(50)    NOT NULL ,
        gender           bool,
        type INT,
 
        log_time datetime NOT NULL 
       
      
        );''')

       #  self.c.execute('''CREATE TABLE IF NOT EXISTS admin
       # (
       #  id_number            CHAR(50)    NOT NULL ,
       #
       #
       #  password        char(20)    NOT NULL,
       #  salt            char(10)  NOT NULL ,
       #  vector          blob        ,
       #  PRIMARY KEY (id_number )
       #           )without rowid;''')
       #  self.c.execute('''CREATE TABLE IF NOT EXISTS admin_log_time
       # (
       #  id_number             CHAR(50)    NOT NULL ,
       #
       #  log_time datetime NOT NULL
       #
       #
       #  );''')

        self.conn.commit()

    def insert_user(self, id_number, user_name,gender, password, img_path, vector,
                    salt):
        self.c.execute(
            "INSERT INTO student (id_number,user_name,gender,password ,img_path ,vector,salt) \
      VALUES (?,?, ?, ? , ?,?,?)",
            (id_number, user_name,gender, password, img_path, vector, salt))
        self.conn.commit()

    def delete(self, id):
        self.c.execute("delete from student where id_number = {0}".format(id))
        self.conn.commit()