tubearchivist/tubearchivist-server
1
0
Fork 0
mirror of https://github.com/tubearchivist/tubearchivist-server.git synced 2024-11-21 19:30:12 +00:00
Code Issues Projects Releases Wiki Activity

call INCR on version check request

Browse Source
This commit is contained in:
simon 2022-12-22 10:41:33 +07:00
parent 35a90941ff
commit 7d48b7f78f
No known key found for this signature in database
GPG Key ID: 2C15AA5E89985DD4
2 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tubearchivist/web/src/ta_redis.py
View File

@ -86,3 +86,16 @@ class TaskHandler(RedisBase):
def set_pub(self): def set_pub(self):
"""set message to pub""" """set message to pub"""
self.conn.publish(self.key, self.repo_conf.get("gh_repo")) self.conn.publish(self.key, self.repo_conf.get("gh_repo"))
class VersionCheckCounter(RedisBase):
"""count requests to version check API endpoint"""
def __init__(self):
super().__init__()
self.timestamp = datetime.now().strftime("%Y%m%d")
self.key = f"{self.NAME_SPACE}versioncounter:{self.timestamp}"
def increase(self):
"""increase counter by one"""
self.conn.execute_command("INCR", self.key)

2
tubearchivist/web/views.py
View File

@ -5,6 +5,7 @@ from os import environ
from apscheduler.schedulers.background import BackgroundScheduler from apscheduler.schedulers.background import BackgroundScheduler
from flask import Flask, render_template, jsonify, request, redirect from flask import Flask, render_template, jsonify, request, redirect
from src.api_docker import run_docker_backup from src.api_docker import run_docker_backup
from src.ta_redis import VersionCheckCounter
from src.webhook_docker import DockerHook from src.webhook_docker import DockerHook
from src.webhook_github import GithubBackup, GithubHook from src.webhook_github import GithubBackup, GithubHook
import markdown import markdown
@ -44,6 +45,7 @@ def discord_redirect():
def release(release_id): def release(release_id):
"""api release""" """api release"""
result = GithubBackup(release_id).get_tag() result = GithubBackup(release_id).get_tag()
VersionCheckCounter().increase()
return jsonify(result) return jsonify(result)