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)