-
Notifications
You must be signed in to change notification settings - Fork 0
/
seed_data.py
75 lines (56 loc) · 2.04 KB
/
seed_data.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
from faker import Faker
from database import SessionLocal
from models import Hospital, Doctor, Patient, GenderEnum
import random
from insert_data import insert_hospital, insert_doctor, insert_patient
from database_utils import get_doctor_by_id, get_hospital_by_id, get_patient_by_id
import logging
# Initialize Faker
fake = Faker()
def create_fake_hospital(num):
for _ in range(num):
insert_hospital(
name=fake.company(),
address=fake.address(),
)
def create_fake_doctor(num):
hospitals = SessionLocal().query(Hospital).all()
for _ in range(num):
insert_doctor(
name=fake.name(),
gender=random.choice(list(GenderEnum)),
email=fake.email(),
hospital_id=random.choice(hospitals).id,
)
SessionLocal().close()
def create_fake_patient(num):
for _ in range(num):
insert_patient(
name=fake.name(),
gender=random.choice(list(GenderEnum)),
birthday=fake.date(),
)
SessionLocal().close()
def associate_patients_randomly():
try:
with SessionLocal() as session:
patients = session.query(Patient).all()
doctors = session.query(Doctor).all()
for patient in patients:
patient = get_patient_by_id(session, patient.id)
doctor = get_doctor_by_id(session, random.choice(doctors).id)
hospital = get_hospital_by_id(session, doctor.hospital_id)
patient.doctors.append(doctor)
patient.hospitals.append(hospital)
session.add(patient)
session.commit()
logging.info(
f"Patient '{patient.name}' associated with Doctor '{doctor.name}' and Hospital '{hospital.name}'."
)
except Exception as e:
logging.error(f"Error associating patient: {e}")
if __name__ == "__main__":
# create_fake_hospital(10)
# create_fake_doctor(20)
# create_fake_patient(30)
associate_patients_randomly()