mirror of
https://github.com/BreizhHardware/ntfy_alerts.git
synced 2026-01-18 16:37:28 +01:00
Félix MARQUET
e8eb8d18d2
- Refactored `send_gotify.py`, `send_ntfy.py`, and `send_discord.py` to use separate database connections for each function. - Added `get_db_connection` function to create a new database connection. - Updated `github_send_to_gotify`, `docker_send_to_gotify`, `github_send_to_ntfy`, `docker_send_to_ntfy`, `github_send_to_discord`, and `docker_send_to_discord` functions to use the new `get_db_connection` function. - Modified `ntfy.py` to use threading for calling notification functions simultaneously.
99 lines
4.0 KiB
Python
99 lines
4.0 KiB
Python
import requests
|
|
import sqlite3
|
|
import logging
|
|
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
|
|
)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
def get_db_connection():
|
|
return sqlite3.connect("/github-ntfy/ghntfy_versions.db", check_same_thread=False)
|
|
|
|
def github_send_to_gotify(releases, token, url):
|
|
conn = get_db_connection()
|
|
cursor = conn.cursor()
|
|
url = url + "/message"
|
|
url = url + "?token=" + token
|
|
for release in releases:
|
|
app_name = release["repo"].split("/")[-1] # Getting the application name from the repo
|
|
version_number = release["tag_name"] # Getting the version number
|
|
app_url = release["html_url"] # Getting the application URL
|
|
changelog = release["changelog"] # Getting the changelog
|
|
release_date = release["published_at"] # Getting the release date
|
|
release_date = release_date.replace("T", " ").replace("Z", "") # Formatting the release date
|
|
|
|
# Checking if the version has changed since the last time
|
|
cursor.execute(
|
|
"SELECT version FROM versions WHERE repo=?",
|
|
(app_name,),
|
|
)
|
|
previous_version = cursor.fetchone()
|
|
if previous_version and previous_version[0] == version_number:
|
|
logger.info(f"The version of {app_name} has not changed. No notification sent.")
|
|
continue # Move on to the next application
|
|
|
|
message = f"New version: {version_number}\nFor: {app_name}\nPublished on: {release_date}\nChangelog:\n{changelog}\n{app_url}"
|
|
# Updating the previous version for this application
|
|
cursor.execute(
|
|
"INSERT OR REPLACE INTO versions (repo, version, changelog) VALUES (?, ?, ?)",
|
|
(app_name, version_number, changelog),
|
|
)
|
|
conn.commit()
|
|
|
|
content = {
|
|
"title": f"New version for {app_name}",
|
|
"message": message,
|
|
"priority": "2",
|
|
}
|
|
response = requests.post(url, json=content)
|
|
if response.status_code == 200:
|
|
logger.info(f"Message sent to Gotify for {app_name}")
|
|
continue
|
|
else:
|
|
logger.error(f"Failed to send message to Gotify. Status code: {response.status_code}")
|
|
|
|
|
|
def docker_send_to_gotify(releases, token, url):
|
|
conn = get_db_connection()
|
|
cursor = conn.cursor()
|
|
url = url + "/message"
|
|
url = url + "?token=" + token
|
|
for release in releases:
|
|
app_name = release["repo"].split("/")[-1] # Getting the application name from the repo
|
|
digest_number = release["digest"]
|
|
app_url = release["html_url"] # Getting the application URL
|
|
release_date = release["published_at"] # Getting the release date
|
|
release_date = release_date.replace("T", " ").replace("Z", "") # Formatting the release date
|
|
|
|
# Checking if the version has changed since the last time
|
|
cursor.execute(
|
|
"SELECT digest FROM docker_versions WHERE repo=?",
|
|
(app_name,),
|
|
)
|
|
previous_digest = cursor.fetchone()
|
|
if previous_digest and previous_digest[0] == digest_number:
|
|
logger.info(f"The digest of {app_name} has not changed. No notification sent.")
|
|
continue # Move on to the next application
|
|
|
|
message = f"New version: {digest_number}\nFor: {app_name}\nPublished on: {release_date}\n{app_url}"
|
|
# Updating the previous digest for this application
|
|
cursor.execute(
|
|
"INSERT OR REPLACE INTO docker_versions (repo, digest) VALUES (?, ?, ?)",
|
|
(app_name, digest_number),
|
|
)
|
|
conn.commit()
|
|
|
|
content = {
|
|
"title": f"New version for {app_name}",
|
|
"message": message,
|
|
"priority": "2",
|
|
}
|
|
response = requests.post(url, json=content)
|
|
if response.status_code == 200:
|
|
logger.info(f"Message sent to Gotify for {app_name}")
|
|
continue
|
|
else:
|
|
logger.error(f"Failed to send message to Gotify. Status code: {response.status_code}")
|