Пример #1
0
 def __init__(self,
              file_path,
              start_date=None,
              end_date=None,
              today=False,
              this_week=False,
              this_month=False):
     self.today = today
     if today:
         self.start_date = get_today()
         self.end_date = get_tomorrow()
     elif this_week:
         self.start_date = get_last_week()
         self.end_date = get_tomorrow()
     elif this_month:
         self.start_date = get_last_month()
         self.end_date = get_tomorrow()
     else:
         self.start_date = start_date
         self.end_date = end_date
     self.components_total = read_calendar(file_path, self.start_date,
                                           self.end_date)
     self.name = file_path.split('/')[-1]
     self.time_delta_dict = get_time_by_date(self.components_total)
     if len(self.components_total) == 0:
         no_cal_data_warning = 'No tasks found in {} calendar for period between {} and {}. Did you remember to export an up-to-date version of {} to timetrack/data/calendars/?'.format(
             self.name, self.start_date, self.end_date, self.name)
         warnings.warn(no_cal_data_warning)
Пример #2
0
def pbf_download_needed(pbf_file_with_date, md5_file_with_date, pgosm_date):
    """Decides if the PBF/MD5 files need to be downloaded.

    Parameters
    -------------------------------
    pbf_file_with_date : str
    md5_file_with_date : str

    Returns
    --------------------------
    download_needed : bool
    """
    logger = logging.getLogger('pgosm-flex')
    # If the PBF file exists, check for the MD5 file too.
    if os.path.exists(pbf_file_with_date):
        logger.info(f'PBF File exists {pbf_file_with_date}')

        if os.path.exists(md5_file_with_date):
            logger.info('PBF & MD5 files exist.  Download not needed')
            download_needed = False
        else:
            if pgosm_date == helpers.get_today():
                print('PBF for today available but not MD5... download needed')
                download_needed = True
            else:
                err = f'Missing MD5 file for {pgosm_date}. Cannot validate.'
                logger.error(err)
                raise FileNotFoundError(err)
    else:
        if not pgosm_date == helpers.get_today():
            err = f'Missing PBF file for {pgosm_date}. Cannot proceed.'
            logger.error(err)
            raise FileNotFoundError(err)

        logger.info('PBF file not found locally. Download required')
        download_needed = True

    return download_needed
Пример #3
0
def init_socket_wrapper(socket, config):
    return SocketWrapper(socket, config.SOCKET_NAMESPACE)


def init_tryer(app):
    return Tryer(app.logger)


def init_voter(socket_wrapper):
    return Voter(socket_wrapper)


##############
# Initialize #
##############

config = init_config()
application = init_app(config)
limiter = init_limiter()
searcher = init_searcher(config)
socket = init_socket(application)
socket_wrapper = init_socket_wrapper(socket, config)
today = get_today()
tryer = init_tryer(application)
voter = init_voter(socket_wrapper)


def run():
    socket.run(application, port=config.PORT)
Пример #4
0
    '--layerset-path',
    required=False,
    help=
    'Custom path to load layerset INI from. Custom paths should be mounted to Docker via docker run -v ...'
)
@click.option(
    '--language',
    default=None,
    envvar="PGOSM_LANGUAGE",
    help=
    "Set default language in loaded OpenStreetMap data when available.  e.g. 'en' or 'kn'."
)
@click.option(
    '--pgosm-date',
    required=False,
    default=helpers.get_today(),
    envvar="PGOSM_DATE",
    help=
    "Date of the data in YYYY-MM-DD format. If today (default), automatically downloads when files not found locally. Set to historic date to load locally archived PBF/MD5 file, will fail if both files do not exist."
)
@click.option('--schema-name',
              required=False,
              default='osm',
              help="Change the final schema name, defaults to 'osm'.")
@click.option(
    '--skip-dump',
    default=False,
    is_flag=True,
    help=
    'Skips the final pg_dump at the end. Useful for local testing when not loading into more permanent instance.'
)
Пример #5
0
# random forest
model_rf = rfr(random_state=1)
model_rf.fit(train_X, train_y)
get_mae(model_rf, val_X, val_y)

# xgboost
model_xgb = xgb(random_state=1, n_estimators=10000, learning_rate=0.01)
model_xgb.fit(train_X,
              train_y,
              early_stopping_rounds=10,
              eval_set=[(val_X, val_y)],
              verbose=False)
get_mae(model_xgb, val_X, val_y)

# make predictions for today
year, month, day = get_today()
om = 30.15
od = 23.96
r = 727
rm = 701.35
t = 6.45
ts = 1.79
td = 1.6

d = {
    'Year': [year],
    'Month': [month],
    'Date': [day],
    'OM': om,
    'OD': od,
    'R': r,
Пример #6
0
 def __init__(self, text, date=helpers.get_today(), time=helpers.get_now()):
     self.date = date
     self.time = time
     self.text = text