mirror of
https://github.com/ada-dmitry/oop_ada.git
synced 2026-09-24 00:00:21 +00:00
133 lines
4.2 KiB
Python
133 lines
4.2 KiB
Python
import psycopg2
|
|
|
|
class data_base:
|
|
def __init__(self):
|
|
self.db_params = {
|
|
"dbname": "postgres",
|
|
"user": "myuser",
|
|
"password": "qwerty",
|
|
"host": "localhost"
|
|
}
|
|
|
|
def connect(self):
|
|
return psycopg2.connect(**self.db_params)
|
|
|
|
def close(self, conn, cur):
|
|
if cur:
|
|
cur.close()
|
|
if conn:
|
|
conn.close()
|
|
|
|
|
|
class patient:
|
|
def __init__(self, pat_id, pat_fio, pat_born, medcard_id, db):
|
|
self.pat_id = pat_id
|
|
self.pat_fio = pat_fio
|
|
self.pat_born = pat_born
|
|
self.medcardc_id = medcard_id
|
|
self.db = db
|
|
|
|
def add2db(self):
|
|
conn = None
|
|
cur = None
|
|
try:
|
|
conn = self.db.connect()
|
|
cur = conn.cursor()
|
|
cur.execute("INSERT INTO patient_db (pat_id, pat_fio, pat_born, medcard_id) VALUES (%s, %s, %s, %s)",\
|
|
(self.pat_id, self.pat_fio, self.pat_born, self.medcardc_id))
|
|
conn.commit()
|
|
finally:
|
|
self.db.close(conn, cur)
|
|
|
|
def note_check(self):
|
|
conn = None
|
|
cur = None
|
|
try:
|
|
conn = self.db.connect()
|
|
cur = conn.cursor()
|
|
cur.execute("SELECT * FROM appointment WHERE pat_id = %s", (self.pat_id))
|
|
conn.commit()
|
|
finally:
|
|
self.db.close(conn, cur)
|
|
|
|
class doctor:
|
|
def __init__(self, doc_id, doc_fio, doc_specialization, doc_room, db):
|
|
self.doc_id = doc_id
|
|
self.doc_fio = doc_fio
|
|
self.doc_specialization = doc_specialization
|
|
self.doc_room = doc_room
|
|
self.db = db
|
|
|
|
def add2db(self):
|
|
conn = None
|
|
cur = None
|
|
try:
|
|
conn = self.db.connect()
|
|
cur = conn.cursor()
|
|
cur.execute("INSERT INTO doctors_db (doc_id, doc_fio, doc_specialization, doc_room) VALUES (%s, %s, %s, %s)", \
|
|
(self.doc_id, self.doc_fio, self.doc_specialization, self.doc_room))
|
|
conn.commit()
|
|
finally:
|
|
self.db.close(conn, cur)
|
|
|
|
def workload_check(self, timetable):
|
|
conn = None
|
|
cur = None
|
|
try:
|
|
conn = self.db.connect()
|
|
cur = conn.cursor()
|
|
if (timetable.loanAmount <= 50000):
|
|
# что вместо апдейта?
|
|
cur.execute("UPDATE loan_applications SET status = %s WHERE application_id = %s", ('approved', timetable.applicationID))
|
|
else:
|
|
cur.execute("UPDATE loan_applications SET status = %s WHERE application_id = %s", ('rejected', timetable.applicationID))
|
|
conn.commit()
|
|
finally:
|
|
self.db.close(conn, cur)
|
|
|
|
class MedWorker:
|
|
def __init__(self, mw_id, mw_fio, mw_born, mw_education, db):
|
|
self.mw_id = mw_id
|
|
self.mw_fio = mw_fio
|
|
self.mw_born = mw_born
|
|
self.db = db
|
|
|
|
class DocAppointment:
|
|
def __init__(self, appointment_id, patient, status, db):
|
|
self.appointment_id = appointment_id
|
|
self.patient = patient
|
|
self.status = status
|
|
self.db = db
|
|
|
|
def addToDatabase(self):
|
|
conn = None
|
|
cur = None
|
|
try:
|
|
conn = self.database.connect()
|
|
cur = conn.cursor()
|
|
cur.execute("INSERT INTO loan_applications (client_id, loan_amount, status) VALUES (%s, %s, %s)", (self.client.clientID, self.loanAmount, self.status))
|
|
conn.commit()
|
|
finally:
|
|
self.database.close(conn, cur)
|
|
|
|
def del_appoint(self):
|
|
conn = None
|
|
cur = None
|
|
try:
|
|
conn = self.database.connect()
|
|
cur = conn.cursor()
|
|
cur.execute("DELETE FROM appointment WHERE appointment_id = %s", self.appointment_id)
|
|
conn.commit()
|
|
finally:
|
|
self.database.close(conn, cur)
|
|
|
|
def add_appoint(self):
|
|
conn = None
|
|
cur = None
|
|
try:
|
|
conn = self.database.connect()
|
|
cur = conn.cursor()
|
|
cur.execute("INSERT INTO appointment (appointment_id, patient, status) VALUES (%s, %s, %s)", self.appointment_id, self.patient, self.status)
|
|
conn.commit()
|
|
finally:
|
|
self.database.close(conn, cur) |