mirror of
https://github.com/clearlinux/telemetrics-backend.git
synced 2026-09-05 13:21:28 +00:00
48 lines
1.0 KiB
Python
48 lines
1.0 KiB
Python
# Copyright (C) 2020 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import os
|
|
import psycopg2
|
|
from processing.main import (
|
|
get_latest_id,
|
|
connect,
|
|
PROCESSED_VIEW,)
|
|
|
|
|
|
LAST_PROCESSED = """
|
|
CREATE OR REPLACE VIEW {} AS
|
|
SELECT MAX(id) AS last_id FROM records WHERE processed = True
|
|
""".format(PROCESSED_VIEW)
|
|
|
|
|
|
class Conf():
|
|
db_name = os.environ['POSTGRES_DB']
|
|
db_host = os.environ['POSTGRES_HOSTNAME']
|
|
db_user = os.environ['POSTGRES_USER']
|
|
db_passwd = os.environ['POSTGRES_PASSWORD']
|
|
|
|
|
|
def check_processed_id(conn):
|
|
cursor = conn.cursor()
|
|
try:
|
|
get_latest_id(cursor)
|
|
except psycopg2.errors.UndefinedTable:
|
|
cursor.execute(LAST_PROCESSED)
|
|
finally:
|
|
cursor.close()
|
|
|
|
|
|
def main():
|
|
try:
|
|
conn = connect(Conf())
|
|
check_processed_id(conn)
|
|
conn.close()
|
|
except psycopg2.OperationalError as op_error:
|
|
print("could not connect to server: Connection refused")
|
|
exit(1)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|
|
|
|
# vi: ts=4 et sw=4 sts=4
|