mirror of
https://github.com/GSA/notifications-api.git
synced 2026-07-15 19:01:50 -04:00
fix wheel and click_datetime
This commit is contained in:
@@ -8,7 +8,7 @@ from os import getenv
|
||||
|
||||
import click
|
||||
import flask
|
||||
from click_datetime import Datetime as click_dt
|
||||
# from click_datetime import Datetime as click_dt
|
||||
from faker import Faker
|
||||
from flask import current_app, json
|
||||
from sqlalchemy import and_, select, text, update
|
||||
@@ -260,50 +260,50 @@ def bulk_invite_user_to_service(file_name, service_id, user_id, auth_type, permi
|
||||
file.close()
|
||||
|
||||
|
||||
@notify_command(name="archive-jobs-created-between-dates")
|
||||
@click.option(
|
||||
"-s",
|
||||
"--start_date",
|
||||
required=True,
|
||||
help="start date inclusive",
|
||||
type=click_dt(format="%Y-%m-%d"),
|
||||
)
|
||||
@click.option(
|
||||
"-e",
|
||||
"--end_date",
|
||||
required=True,
|
||||
help="end date inclusive",
|
||||
type=click_dt(format="%Y-%m-%d"),
|
||||
)
|
||||
def update_jobs_archived_flag(start_date, end_date):
|
||||
current_app.logger.info(
|
||||
f"Archiving jobs created between {start_date} to {end_date}"
|
||||
)
|
||||
# @notify_command(name="archive-jobs-created-between-dates")
|
||||
# @click.option(
|
||||
# "-s",
|
||||
# "--start_date",
|
||||
# required=True,
|
||||
# help="start date inclusive",
|
||||
# type=click_dt(format="%Y-%m-%d"),
|
||||
# )
|
||||
# @click.option(
|
||||
# "-e",
|
||||
# "--end_date",
|
||||
# required=True,
|
||||
# help="end date inclusive",
|
||||
# type=click_dt(format="%Y-%m-%d"),
|
||||
# )
|
||||
# def update_jobs_archived_flag(start_date, end_date):
|
||||
# current_app.logger.info(
|
||||
# f"Archiving jobs created between {start_date} to {end_date}"
|
||||
# )
|
||||
|
||||
process_date = start_date
|
||||
total_updated = 0
|
||||
# process_date = start_date
|
||||
# total_updated = 0
|
||||
|
||||
while process_date < end_date:
|
||||
start_time = utc_now()
|
||||
sql = """update
|
||||
jobs set archived = true
|
||||
where
|
||||
created_at >= (date :start + time '00:00:00')
|
||||
and created_at < (date :end + time '00:00:00')
|
||||
"""
|
||||
result = db.session.execute(
|
||||
text(sql), {"start": process_date, "end": process_date + timedelta(days=1)}
|
||||
)
|
||||
db.session.commit()
|
||||
current_app.logger.info(
|
||||
f"jobs: --- Completed took {datetime.now() - start_time}ms. Archived "
|
||||
f"{result.rowcount} jobs for {process_date}"
|
||||
)
|
||||
# while process_date < end_date:
|
||||
# start_time = utc_now()
|
||||
# sql = """update
|
||||
# jobs set archived = true
|
||||
# where
|
||||
# created_at >= (date :start + time '00:00:00')
|
||||
# and created_at < (date :end + time '00:00:00')
|
||||
# """
|
||||
# result = db.session.execute(
|
||||
# text(sql), {"start": process_date, "end": process_date + timedelta(days=1)}
|
||||
# )
|
||||
# db.session.commit()
|
||||
# current_app.logger.info(
|
||||
# f"jobs: --- Completed took {datetime.now() - start_time}ms. Archived "
|
||||
# f"{result.rowcount} jobs for {process_date}"
|
||||
# )
|
||||
|
||||
process_date += timedelta(days=1)
|
||||
# process_date += timedelta(days=1)
|
||||
|
||||
total_updated += result.rowcount
|
||||
current_app.logger.info(f"Total archived jobs = {total_updated}")
|
||||
# total_updated += result.rowcount
|
||||
# current_app.logger.info(f"Total archived jobs = {total_updated}")
|
||||
|
||||
|
||||
@notify_command(name="populate-organizations-from-file")
|
||||
|
||||
Reference in New Issue
Block a user