# -*- coding: UTF-8 -*-

# 命名方式为表名_操作_字段

import os

import tempfile

import connectDB

from controller import fileController

curosrdiction = connectDB.curosr_diction

database_back_folder = “”

def exec_sql(sqltext):

”’

执行sql，成功返回true

:param sqltext: sql语句

:return:

”’

result = curosrdiction.executescript(sqltext)

curosrdiction.commit()

return result

def backup_database():

”’

备份数据库

:return:

”’

global database_back_folder

database_back_folder = tempfile.mkdtemp()

fileController.copy_file_to_folder(‘resource/sqllite.db’, database_back_folder)

write_bat()

def restore_database():

”’

恢复数据库

:return:

”’

curosrdiction.close()

fileController.copy_file(srcfile=database_back_folder + “\\sqllite.db”, dstfile=’resource/sqllite.db’)

def read_sql_from_text(file_name):

”’

从文本文件中读取sql并且执行，执行失败，就把原来的数据库覆盖回来

:param file_name:

:return:

”’

f = open(file_name, ‘r’)

sql = f.read()

if not exec_sql(sql):

run_bat()

def write_bat():

sql_database_path = os.getcwd() + “\\resource\\sqllite.db”

sql_database_path_shm = os.getcwd() + “\\resource\\sqllite.db-shm”

sql_database_path_wal = os.getcwd() + “\\resource\\sqllite.db-wal”

bat_name = ‘copy.bat’

s1 = ”’@echo off

@ping -n 5 127.1 >nul 2>nul

echo delete database…

del ”’ + sql_database_path + ”’

del ”’ + sql_database_path_shm + ”’

del ”’ + sql_database_path_wal + ”’

echo restore database…

@ping -n 5 127.1 >nul 2>nul

copy ”’ + database_back_folder + “\\sqllite.db ” + sql_database_path + ”’

echo restore finish

pause”’

f = open(bat_name, ‘w’)

f.write(s1)

f.close()

def run_bat():

”’

运行bat

:return:

”’

os.system(‘start copy.bat’)

def update_sql():

”’

开始更新数据库的主入口

:return:

”’

backup_database()

read_sql_from_text(“resource/download/1.2.3_sql.txt”)