import?ply.yacc?as?yacc
from?Lex?import??tokens
import?os
import?re
import?shutil?#刪除非空目錄的函數(shù)?rmtree

root?=?‘DBMS‘
conf?=?‘sys.dat‘
current_db?=?‘‘

#################################################################33
#數(shù)據(jù)庫操作函數(shù)定義

#?檢查已存在的數(shù)據(jù)庫并返回
def?check_db（）:
????if?not?os.path.exists（root）:
????????os.mkdir（root）
????if?not?os.path.exists（os.path.join（root?conf））:
????????open（os.path.join（root?conf）?‘w‘）
????????return?[]
????fp?=?open（os.path.join（root?conf）?‘r‘）
????names?=?[name.strip（）?for?name?in?fp.readlines（）?if?name]
????fp.close（）
????return?names

#?創(chuàng)建數(shù)據(jù)庫
def?create_db（name）:
????exisNames?=?check_db（）
????if?name?in?exisNames:
????????return?False
????fp?=?open（os.path.join（root?conf）?‘a(chǎn)‘）
????fp.write（name?+?‘\n‘）
????fp.close（）
????os.mkdir（os.path.join（root?name））
????open（os.sep.join（[root?name?conf]）?‘w‘）
????return?True

#?顯示所有數(shù)據(jù)庫
def?show_db（names）:
????print（names）

#?刪除某個數(shù)據(jù)庫
def?drop_db（name）:
????names?=?check_db（）
????if?name?not?in?names:?return?False
????names.remove（name）
????fp?=?open（os.path.join（root?conf）?‘w‘）
????for?s?in?names:
????????fp.write（s?+?‘\n‘）
????fp.close（）
????shutil.rmtree（os.path.join（root?name））#能夠刪除非空目錄
????return?True
######################################################
def?check_tb（）:
????global?current_db
????if?current_db?==?‘‘:
????????print（‘未選中數(shù)據(jù)庫！‘）
????????return?False
????tb_root?=?os.sep.join（[root?current_db]）
????if?not?os.path.exists（os.path.join（tb_root?conf））:
????????print（‘找不到配置文件‘os.path.join（tb_root?conf））
????????return
????tbs?=?{}
????fp?=?open（os.path.join（tb_root?conf）?‘r‘）
????for?line?in?fp.readlines（）:
????????s?=?line.strip（）.split（）
????????tmp?=?{‘name‘:?s[2]}
????????tmp[‘prop‘]?=?{‘type‘:s[3]}
????????if?len（s）!=4:
????????????tmp[‘prop‘][‘len‘]?=?int（s[4]）
????????if?int（s[1]）?==?1:
????????????tbs[s[0]]?=?[]
????????tbs[s[0]].append（tmp）
????fp.close（）
????#檢查配置文件中的表格是否存在
????tmp?=?list（tbs.keys（））
????for?tb_name?in?tmp:
????????if?not?os.path.exists（os.path.join（tb_root?tb_name+‘.txt‘））:
????????????tbs.pop（tb_name）
????#檢查當(dāng)前目錄中的txt文件是否都在dat中聲明
????tb_names?=?[t[:-4]?for?t?in?os.listdir（tb_root）?if?t[-3:]?==?‘txt‘]
????for?t?in?tb_names:
????????if?tbs.get（t）==None:
????????????os.remove（os.path.join（tb_root?t+‘.txt‘））
????write_tb_conf（tbs）

????#?print（tbs）
????return?tbs

#?寫table的conf的函數(shù)
def?write_tb_conf（tbs）:
????if?current_db?==?‘‘:
????????print（‘未選中數(shù)據(jù)庫！‘）
????????return?False
????tb_root?=?os.sep.join（[root?current_db]）
????fp?=?open（os.path.join（tb_root?conf）?‘w‘）
????for?tb_name?tb_elements?in?tbs.items（）:
????????for?i?in?range（len（tb_elements））:
????????????if?tb_elements[i][‘prop‘].get（‘len‘）:
????????????????fp.write（‘%s?%d?%s?%s?%d\n‘?%?（tb_name?i?+?1?tb_elements[i][‘name‘]?\
???????????????????????????????????????????????tb_elements[i][‘prop‘][‘type‘]?\
???????????????????????????????????????????????tb_elements[i]