def import_logfile(ogn_logfile, reference_date, logfile='main.log', loglevel='INFO'): """Import OGN-data from ogn-log-files <arg: ogn-logfile, reference_date>. Reference date must be given in YYYY-MM-DD.""" # Check if filename exists try: f = open(ogn_logfile, 'r') except: print('\nError reading ogn-logfile:', ogn_logfile) return try: reference_date = datetime.strptime(reference_date, "%Y-%m-%d") except: print('\nError in reference_date argument', reference_date) return # Enable logging log_handlers = [logging.StreamHandler()] if logfile: log_handlers.append(logging.FileHandler(logfile)) logging.basicConfig(format=logging_formatstr, level=loglevel, handlers=log_handlers) beacons = list() print('Start importing ogn-logfile') for line in f: beacon = message_to_beacon(line, reference_date=reference_date) if beacon is not None: beacons.append(beacon) session.bulk_save_objects(beacons) f.close() logging.shutdown()
def import_airports(path='tests/SeeYou.cup'): """Import airports from a ".cup" file""" print("Import airports from '{}'...".format(path)) airports = get_airports(path) session.bulk_save_objects(airports) session.commit() print("Imported {} airports.".format(len(airports)))