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()