ufatih/vigil
1
0
Fork 0
Code Issues Pull Requests Activity

Fixed it

Browse Source
This commit is contained in:
Uthman Fatih 2025-10-15 08:14:18 +01:00
parent fba3271f54
commit 9b561d6983
1 changed files with 19 additions and 41 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

60
main.py
View File

@ -43,27 +43,11 @@ def now(receiver_email: str):
generate_email(receiver_email) generate_email(receiver_email)
def generate_email(receiver_email: str): def generate_email(receiver_email: str):
#services.apt.require_root() services.apt.require_root()
#if not services.apt.detect_apt(): if not services.apt.detect_apt():
# print("Apt not found on this system.") print("Apt not found on this system.")
# exit(1) exit(1)
#updates = services.apt.check_updates() updates = services.apt.check_updates()
# TODO: Remove this after testing
updates = {
"openssl": {
"security": True,
"installed_version": "1.0.0",
"latest_version": "1.1.0",
"repo": "main",
},
"vim": {
"security": False,
"installed_version": "2.0.0",
"latest_version": "2.1.0",
"repo": "universe",
}
}
# Get how many security updates are available # Get how many security updates are available
security_updates = 0 security_updates = 0
@ -76,8 +60,9 @@ def generate_email(receiver_email: str):
# Get how many general updates are available # Get how many general updates are available
general_updates = total_updates - security_updates general_updates = total_updates - security_updates
if general_updates == 0: # Check if there are any updates at all
print("No general updates available.") if total_updates == 0:
print("No updates available.")
return return
# Get system hostname # Get system hostname
@ -86,16 +71,12 @@ def generate_email(receiver_email: str):
# Create email subject # Create email subject
subject = f"{hostname}> {security_updates} security updates, {general_updates} general updates available" subject = f"{hostname}> {security_updates} security updates, {general_updates} general updates available"
chunk = f'' # define chunk # Build security updates section
# Iterate over each security package and create a list of updates in html
security_chunks = "" security_chunks = ""
for package in updates: for package in updates:
if not updates[package]["security"]: if not updates[package]["security"]:
continue continue
if security_updates == 0:
continue
chunk = f''' chunk = f'''
<tr> <tr>
<td style="border: 1px solid #ddd; padding: 8px;">{package}</td> <td style="border: 1px solid #ddd; padding: 8px;">{package}</td>
@ -104,10 +85,11 @@ def generate_email(receiver_email: str):
<td style="border: 1px solid #ddd; padding: 8px;">{updates[package]["repo"]}</td> <td style="border: 1px solid #ddd; padding: 8px;">{updates[package]["repo"]}</td>
</tr> </tr>
''' '''
security_chunks += chunk
security_chunks += chunk security = ""
if security_updates > 0:
security = f''' security = f'''
<h1 style="font-family: Arial, sans-serif;">Security</h1> <h1 style="font-family: Arial, sans-serif;">Security</h1>
<table style="border-collapse: collapse; width: 100%; font-family: Arial, sans-serif;"> <table style="border-collapse: collapse; width: 100%; font-family: Arial, sans-serif;">
<tr> <tr>
@ -120,15 +102,12 @@ def generate_email(receiver_email: str):
</table> </table>
''' '''
# Iterate over each general package and create a list of updates in html # Build general updates section
general_chunks = "" general_chunks = ""
for package in updates: for package in updates:
if updates[package]["security"]: if updates[package]["security"]:
continue continue
if general_updates == 0:
continue
chunk = f''' chunk = f'''
<tr> <tr>
<td style="border: 1px solid #ddd; padding: 8px;">{package}</td> <td style="border: 1px solid #ddd; padding: 8px;">{package}</td>
@ -137,10 +116,11 @@ def generate_email(receiver_email: str):
<td style="border: 1px solid #ddd; padding: 8px;">{updates[package]["repo"]}</td> <td style="border: 1px solid #ddd; padding: 8px;">{updates[package]["repo"]}</td>
</tr> </tr>
''' '''
general_chunks += chunk
general_chunks += chunk general = ""
if general_updates > 0:
general = f''' general = f'''
<h1 style="font-family: Arial, sans-serif;">General</h1> <h1 style="font-family: Arial, sans-serif;">General</h1>
<table style="border-collapse: collapse; width: 100%; font-family: Arial, sans-serif;"> <table style="border-collapse: collapse; width: 100%; font-family: Arial, sans-serif;">
<tr> <tr>
@ -153,9 +133,7 @@ def generate_email(receiver_email: str):
</table> </table>
''' '''
html = "" html = security + general
html += security
html += general
services.mail.send_email(receiver_email, subject, html) services.mail.send_email(receiver_email, subject, html)