diff --git a/daemon.py b/daemon.py index 721434f..7fbcd9b 100644 --- a/daemon.py +++ b/daemon.py @@ -65,6 +65,7 @@ print("[client]" + str(msg.payload)) def verify_card(msg): + DB_connection_check() card_number = '{:010d}'.format(int(msg.payload)) status = AuthStatus.EXCEPTION try: diff --git a/db_models.py b/db_models.py index 13f209c..21595b0 100644 --- a/db_models.py +++ b/db_models.py @@ -26,11 +26,7 @@ def DB_Init(): mysql_db.connect() #连接数据库 mysql_db.create_tables([AccountInfo, AccessRecords], safe=True) - # AccountInfo.create( - # realname = 'good one', - # studnum = '2017011234', - # cardnum = '0000080801', - # cardtype = 2, - # userid = 'test', - # ).save() + +def DB_connection_check(): + mysql_db.get_conn().ping(True)