| 11 | self.acc_no = self._get_last_acc_no() + 1 |
| 12 | |
| 13 | def _setup_tables(self): |
| 14 | self.cur.execute(""" |
| 15 | CREATE TABLE IF NOT EXISTS bank ( |
| 16 | acc_no INTEGER PRIMARY KEY, |
| 17 | name TEXT, |
| 18 | age INTEGER, |
| 19 | address TEXT, |
| 20 | balance INTEGER, |
| 21 | account_type TEXT, |
| 22 | mobile_number TEXT |
| 23 | ) |
| 24 | """) |
| 25 | self.cur.execute(""" |
| 26 | CREATE TABLE IF NOT EXISTS staff ( |
| 27 | name TEXT, |
| 28 | pass TEXT, |
| 29 | salary INTEGER, |
| 30 | position TEXT |
| 31 | ) |
| 32 | """) |
| 33 | self.cur.execute("CREATE TABLE IF NOT EXISTS admin (name TEXT, pass TEXT)") |
| 34 | self.cur.execute("SELECT COUNT(*) FROM admin") |
| 35 | if self.cur.fetchone()[0] == 0: |
| 36 | self.cur.execute("INSERT INTO admin VALUES (?, ?)", ("admin", "admin123")) |
| 37 | self.conn.commit() |
| 38 | |
| 39 | def _get_last_acc_no(self): |
| 40 | self.cur.execute("SELECT MAX(acc_no) FROM bank") |