tcl_ota_check/upload_logs.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

# pylint: disable=C0111,C0326,C0103

"""Upload contents of logs folder to remote database."""

# curl -v -H "Content-Type: text/plain" --data @test.xml http://example.org/tcl_update_db/

import glob
import os
import sys

import requests

from tcllib import argparser
dpdesc = """
    Uploads contents of logs folder to remote database.
    """
dp = argparser.DefaultParser(__file__, dpdesc)
args = dp.parse_args(sys.argv[1:])
del args


# This is the URL to an installation of https://github.com/mbirth/tcl_update_db
UPLOAD_URL = "https://tclota.birth-online.de/"
LOGS_GLOB = os.path.normpath("logs/*.xml")

headers = {"Content-Type": "text/xml"}

file_list = glob.glob(LOGS_GLOB)
print("Found {} logs to upload.".format(len(file_list)))
for fn in file_list:
    print("Uploading {}…".format(fn), end="", flush=True)
    with open(fn, "rb") as f:
        r = requests.post(UPLOAD_URL, data=f, headers=headers)
    if r.status_code == 200:
        os.remove(fn)
        print(" OK")
    else:
        add_text = ""
        if r.status_code in [413, 406, 412]:
            # File has been rejected by server, another try won't help
            os.remove(fn)
            add_text = " - Please try again later."
        print(" ERROR: HTTP {}{}".format(r.status_code, add_text))

# Mark prds.json as outdated to fetch updated version
print("Mark PRD cache for update…", end="")
try:
    os.utime("prds.json", times=(1, 1))
    print(" OK")
except OSError as e:
    print(" FAILED")