LukeParke/pypistats.org
1
0
Fork 0
mirror of https://github.com/LukeHagar/pypistats.org.git synced 2025-12-09 12:47:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix for event kwargs

Browse Source
This commit is contained in:
crflynn
2018-04-18 23:34:48 -04:00
parent 5e5d0645ed
commit 9cb0cfb9d1
1 changed files with 13 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
pypistats/tasks/pypi.py
View File

@@ -72,7 +72,7 @@ def get_google_credentials():
def get_daily_download_stats(event, context): def get_daily_download_stats(event, context):
"""Get daily download stats for pypi packages from BigQuery.""" """Get daily download stats for pypi packages from BigQuery."""
start = time.time() start = time.time()
env = event["env"] env = event["kwargs"]["env"]
if os.environ.get("ENV", None) is None: if os.environ.get("ENV", None) is None:
load_env_vars(env) load_env_vars(env)
job_config = bigquery.QueryJobConfig() job_config = bigquery.QueryJobConfig()
@@ -81,7 +81,7 @@ def get_daily_download_stats(event, context):
project=os.environ["GOOGLE_PROJECT_ID"], project=os.environ["GOOGLE_PROJECT_ID"],
credentials=credentials credentials=credentials
) )
date = event.get("date", None) date = event["kwargs"].get("date", None)
if date is None: if date is None:
date = str(datetime.date.today() - datetime.timedelta(days=1)) date = str(datetime.date.today() - datetime.timedelta(days=1))
@@ -210,11 +210,11 @@ def update_all_package_stats(event, context):
print("__all__") print("__all__")
start = time.time() start = time.time()
date = event.get("date", None) date = event["kwargs"].get("date", None)
if date is None: if date is None:
date = str(datetime.date.today() - datetime.timedelta(days=1)) date = str(datetime.date.today() - datetime.timedelta(days=1))
env = event["env"] env = event["kwargs"]["env"]
if os.environ.get("ENV", None) is None: if os.environ.get("ENV", None) is None:
load_env_vars(env) load_env_vars(env)
@@ -254,11 +254,11 @@ def update_recent_stats(event, context):
print("recent") print("recent")
start = time.time() start = time.time()
date = event.get("date", None) date = event["kwargs"].get("date", None)
if date is None: if date is None:
date = str(datetime.date.today() - datetime.timedelta(days=1)) date = str(datetime.date.today() - datetime.timedelta(days=1))
env = event["env"] env = event["kwargs"]["env"]
if os.environ.get("ENV", None) is None: if os.environ.get("ENV", None) is None:
load_env_vars(env) load_env_vars(env)
@@ -328,11 +328,11 @@ def purge_old_data(event, context):
age = MAX_RECORD_AGE age = MAX_RECORD_AGE
start = time.time() start = time.time()
date = event.get("date", None) date = event["kwargs"].get("date", None)
if date is None: if date is None:
date = str(datetime.date.today() - datetime.timedelta(days=1)) date = str(datetime.date.today() - datetime.timedelta(days=1))
env = event["env"] env = event["kwargs"]["env"]
if os.environ.get("ENV", None) is None: if os.environ.get("ENV", None) is None:
load_env_vars(env) load_env_vars(env)
@@ -450,11 +450,13 @@ def get_query(date):
if __name__ == "__main__": if __name__ == "__main__":
date = "2018-04-17" date = "2018-04-18"
env = "prod" env = "prod"
event = { event = {
"date": date, "kwargs": {
"env": env, "date": date,
"env": env,
}
} }
context = None context = None
print(date, env) print(date, env)