def get_connections():
connections = {}
for facility in Facility.select():
for transfer in facility.transfers_from:
if facility.get_id() not in connections:
connections[facility.get_id()] = {}
if transfer.to_facility is not None:
connections[facility.get_id()][transfer.to_facility.get_id()] = True
return connections
def main(argv):
db.connect()
try:
Detention.drop_table()
except sqlite3.OperationalError:
pass
Detention.create_table()
with open(argv[1]) as f:
date_format = '%d-%b-%y'
csv_file = DictReader(f)
for line in csv_file:
person = Person.select().where(Person.trac_id == line['TRAC Assigned Identifier for Individual'].strip()).get()
facility = Facility.select().where(Facility.name == line['Detention Facility'].strip()).get()
attrs = {
'person': person,
'facility': facility,
'book_in_date': datetime.strptime(line['Book-In Date'].strip(), date_format).date(),
'book_out_date': datetime.strptime(line['Book-Out Date'].strip(), date_format).date(),
'book_out_reason': line['Reason for Book-Out'].strip(),
}
Detention.create(**attrs)
def export_facilities(filename):
with open(filename, 'w') as f:
data = []
for facility in Facility.select().dicts():
data.append(facility)
json.dump(data, f)
