Files
2020-04-03 11:07:05 -07:00

54 lines
1.2 KiB
Python
Executable File

# Copyright (C) 2020 Intel Corporation
# SPDX-License-Identifier: Apache-2.0
import os
import luigi
import psycopg2
from processing.main import (
connect,
init_guilty_blacklist,
get_latest_id,
get_crashes,
process_crashes,
)
from processing import crash
class GlobalParams(luigi.Config):
db_name = os.environ['POSTGRES_DB']
db_host = os.environ['POSTGRES_HOSTNAME']
db_user = os.environ['POSTGRES_USER']
db_passwd = os.environ['POSTGRES_PASSWORD']
class ProcessCrashes(luigi.Task):
"""
process.py ProcessCrashes --local-scheduler
"""
is_complete = False
def complete(self):
return self.is_complete
def run(self):
db = connect(GlobalParams())
cur = db.cursor()
crash.GUILTY_BLACKLIST = init_guilty_blacklist(cur)
last_id = get_latest_id(cur)
print("# Start processing after record id: {}".format(last_id))
crashes = get_crashes(cur, last_id)
print("# Will process [{}] crashes".format(len(crashes)))
process_crashes(cur, crashes)
print("# Done processing\n")
cur.close()
db.close()
self.is_complete = True
if __name__ == "__main__":
luigi.run()
# vi: ts=4 et sw=4 sts=4