TakeoffPostProcessService.py

import multiprocessing
import sys
from datetime import datetime
from multiprocessing.pool import ThreadPool

from injectable import injectable, Autowired, autowired
import geojson as geojson
from TakeOffLocationService import TakeOffLocationService
import igc2geojson as igc2geojson

import logging

logging.basicConfig(
    level=logging.INFO,
    format="%(asctime)s [%(levelname)s] %(message)s",
    handlers=[
        logging.StreamHandler(sys.stdout)
    ]
)
logger = logging.getLogger("")


@injectable
class TakeoffPostProcessService:
    """ Enrich flight with Take off location"""
    geojson_airports: dict

    @autowired
    def __init__(self,
                 takeoff_location_service: Autowired(TakeOffLocationService)):
        self.takeoff_location_service = takeoff_location_service

    def run_ok(self) -> None:
        data = None
        #with open("C:\\llauner\\src\\geojson\\2020_07_07-tracks.geojson") as file:
        with open("C:\\llauner\\src\\geojson\\2021-tracks.geojson") as file:
            data = geojson.load(file)

        print("Current Time =", datetime.now().strftime("%H:%M:%S"))
        for i, f in enumerate(data.features):
            self.find_takeoff_location(f)

        igc2geojson.dump_feature_collection_to_file("C:\\llauner\\src\\geojson\\out", data)
        print("Current Time =", datetime.now().strftime("%H:%M:%S"))

    def find_takeoff_location(self, f):
        if f.geometry.coordinates:
            takeoff_location = f.geometry.coordinates[0]
            found_takeoff_location = self.takeoff_location_service.get_takeoff_location_lat_lon(takeoff_location[1],
                                                                                                takeoff_location[0])
            if found_takeoff_location is not None:
                f.properties['takeoff'] = found_takeoff_location
                return f"Airport found: {found_takeoff_location}"
        return "Not found !"

    def run(self) -> None:
        data = None
        #with open("C:\\llauner\\src\\geojson\\2020_07_07-tracks.geojson") as file:
        with open("C:\\llauner\\src\\geojson\\2020-tracks.geojson") as file:
            data = geojson.load(file)
        items = data.features

        print("Current Time =", datetime.now().strftime("%H:%M:%S"))
        # create a process pool that uses all cpus
        pool = ThreadPool(6)

        pool.imap(self.find_takeoff_location, items)
        # for result in pool.imap(self.find_takeoff_location, items):
        #     logger.info(result)

        # close the process pool
        pool.close()
        igc2geojson.dump_feature_collection_to_file("C:\\llauner\\src\\geojson\\out", data)
        print("Current Time =", datetime.now().strftime("%H:%M:%S"))