model.py

import folium
import pymongo
from enum import Enum
import logging
from collections import OrderedDict
import datetime as dt

logger = logging.getLogger(__name__)

class Pollutant(Enum):
    NO2 = 8
    SO2 = 1
    PM10 = 5
    O3 = 7
    NOX = 9
    CO = 10
    H2S = 11
    PM25 = 6001 

class DBClient:
    """
    Client for connecting to the mongodb database
    """

    def __init__(self, host, port=27017):
        """
        Create the connection and collections
        """
        self.client = pymongo.MongoClient(f"mongodb://{host}:{port}")
        self.db = self.client["airQmap"]
        self.stations = self.db["stations"]
        if not "stations" in self.db.list_collection_names():
            self.stations.create_index("station_ref", unique=True)
        
        self.mesures = self.db["mesures"]
        if not "mesures" in self.db.list_collection_names():
            self.mesures.create_index("mesure_ref", unique=True)

        self.datasets = self.db["datasets"]

    def insert_mesures(self, mesures):
        """
        Insert without duplicates a list of mesures in the database

        Args:
            mesures (list(dict)): list of mesures to insert
        """
        try :
            self.mesures.insert_many(mesures, ordered=False)
        except :
            logger.debug("failed to insert some mesures due to dupplicated references")

    def insert_stations(self, stations):
        """
        Insert without duplicates a list of mesurement stations in the database

        Args:
            stations (list(dict)): list of stations with their id and coordinates
        """
        try :
            self.stations.insert_many(stations, ordered=False)
        except:
            logger.debug("failed to insert some mesures due to dupplicated references")

    def insert_dataset(self, url):
        """
        Insert newly parsed datasets urls

        Args:
            url (string): url of the inserted dataset
        """
        try :
            self.datasets.insert_one({"url":url})
        except :
            logger.debug("failed to insert dataset url")

    def get_mesures_by_time(self, pollutant):
        """
        Retrieve mesures from the database for a pollutant, and groups them by the time of mesurement in a chronological order

        Args:
            pollutant (string): name of the pollutant

        Returns:
            OrderedDict: dict with datetimes as keys and mesurement lists as values
        """
        times = sorted(self.mesures.distinct("end_mesure", filter={
            "pollutant":pollutant,
            "end_mesure": {
                "$gte": dt.datetime.now() - dt.timedelta(days=1)
            }
        }))
        ordered_mesures = OrderedDict()
        for time in times :
            ordered_mesures[time] = list(self.mesures.find({
                "end_mesure":time,
                "pollutant":pollutant,
            }))
        return ordered_mesures