39 lines
1.9 KiB
Python
39 lines
1.9 KiB
Python
import psycopg2, sys
|
|
sys.stdout.reconfigure(encoding='utf-8')
|
|
conn = psycopg2.connect(host='192.168.110.252', port=15432, dbname='postgresql', user='postgresql', password='Jchl1528', options='-c search_path=healthlink_his')
|
|
cur = conn.cursor()
|
|
|
|
# Check if delete_flag exists on antibiotic_approval
|
|
cur.execute("""SELECT column_name FROM information_schema.columns WHERE table_name='antibiotic_approval' AND column_name='delete_flag'""")
|
|
if cur.fetchone():
|
|
print('antibiotic_approval.delete_flag EXISTS')
|
|
else:
|
|
print('antibiotic_approval.delete_flag MISSING - adding now')
|
|
cur.execute("""ALTER TABLE antibiotic_approval ADD COLUMN IF NOT EXISTS delete_flag CHAR(1) DEFAULT '0'""")
|
|
cur.execute("""COMMENT ON COLUMN antibiotic_approval.delete_flag IS 'delete flag (0=normal,1=deleted)'""")
|
|
cur.execute("""UPDATE antibiotic_approval SET delete_flag = '0' WHERE delete_flag IS NULL""")
|
|
conn.commit()
|
|
print('antibiotic_approval.delete_flag ADDED')
|
|
|
|
# Check prescription_intercept_log
|
|
cur.execute("""SELECT column_name FROM information_schema.columns WHERE table_name='prescription_intercept_log' AND column_name='delete_flag'""")
|
|
if cur.fetchone():
|
|
print('prescription_intercept_log.delete_flag EXISTS')
|
|
else:
|
|
print('prescription_intercept_log.delete_flag MISSING - adding now')
|
|
cur.execute("""ALTER TABLE prescription_intercept_log ADD COLUMN IF NOT EXISTS delete_flag CHAR(1) DEFAULT '0'""")
|
|
conn.commit()
|
|
print('prescription_intercept_log.delete_flag ADDED')
|
|
|
|
# Check sys_audit_log
|
|
cur.execute("""SELECT column_name FROM information_schema.columns WHERE table_name='sys_audit_log' AND column_name='delete_flag'""")
|
|
if cur.fetchone():
|
|
print('sys_audit_log.delete_flag EXISTS')
|
|
else:
|
|
print('sys_audit_log.delete_flag MISSING - adding now')
|
|
cur.execute("""ALTER TABLE sys_audit_log ADD COLUMN IF NOT EXISTS delete_flag CHAR(1) DEFAULT '0'""")
|
|
conn.commit()
|
|
print('sys_audit_log.delete_flag ADDED')
|
|
|
|
cur.close()
|
|
conn.close() |