bubbles/python_modules/finish_cleanup.py

121 lines
4.2 KiB
Python
Raw Normal View History

2023-07-11 06:48:53 +00:00
"""
* Author: "PeppermintOS Team(peppermintosteam@proton.me)
*
* License: SPDX-License-Identifier: GPL-3.0-or-later
*
* Clean up the working folder and move ISO and other files
"""
import os
import shutil
from pathlib import Path
from datetime import date
import time
2023-07-11 06:48:53 +00:00
# Set the home path used regardless the user logged in
BSTRING_ISO_CONFIGS = '~/bubbles/iso_configs'
HOME_FOLDER = str(Path(BSTRING_ISO_CONFIGS).expanduser())
FUSATO_ROOT = '/fusato'
# Set the base destination location for the final ISO
# Default base is /var/www/html/nightly/
BASE_DESTINATION = '/var/www/html/nightly/'
# Set the base ISO Name
BASE_NAME = 'PeppermintOS'
2023-07-11 06:48:53 +00:00
def make_check_sum():
""" Generate the check sum files """
os.chdir(HOME_FOLDER + FUSATO_ROOT)
current_working_directory = os.getcwd().replace('/', '/')
for , _, files in os.walk(current_working_directory):
2023-07-11 06:48:53 +00:00
for fiso in files:
if fiso.endswith('.iso'):
sha = "sha512sum "
ext = "-sha512.checksum"
gen_sum = sha + fiso + ' >' + fiso + ext
os.system(gen_sum)
def process_iso(base, arch, desktop):
"""
Process the ISO for a given base, architecture, and desktop.
"""
current_working_directory = os.listdir(".")
today = str(date.today())
for files in current_working_directory:
if files.endswith('.iso'):
src_iso = files
rdes = f"{BASE_NAME}-{base}_{arch}_{desktop.lower()}{today}.iso"
des_iso =(f"{BASE_DESTINATION}/{base.lower()}{arch} /{desktop.lower()}")
os.rename(src_iso, rdes)
shutil.copy(rdes, des_iso)
elif files.endswith('.checksum'):
src_sum = files
rdes_sum = (
f"{BASE_NAME}{base.capitalize()}_{arch}_{desktop.lower()}"
"-sha512.checksum"
)
des_sum = (f"{BASE_DESTINATION}/{base.lower()}{arch}"/{desktop.lower()}"
)
os.rename(src_sum, rdes_sum)
shutil.copy(rdes_sum, des_sum)
def check_build_type():
""" Decided which ISO needs to be moved for nightly """
# Make sure you are in fustao
os.chdir(HOME_FOLDER + FUSATO_ROOT)
directory_path = HOME_FOLDER + FUSATO_ROOT
file_function_mapping = {
'deb.64xfc': ('deb', '64', 'XFCE'),
'deb.32xfc': ('deb', '32', 'XFCE'),
'deb.armxfc': ('deb', 'arm', 'XFCE'),
'dev.64xfc': ('dev', '64', 'XFCE'),
'dev.32xfc': ('dev', '32', 'XFCE'),
'dev.armxfc': ('dev', 'arm', 'XFCE'),
'deb.64gfb': ('deb', '64', 'GFB'),
'deb.32gfb': ('deb', '32', 'GFB'),
'deb.armgfb': ('deb', 'arm', 'GFB'),
'dev.64gfb': ('dev', '64', 'GFB'),
'dev.32gfb': ('dev', '32', 'GFB'),
'dev.armgfb': ('dev', 'arm', 'GFB'),
}
def check_build():
""" Check what build is there and call the correct process function """
for file_name, params in file_function_mapping.items():
file_path = os.path.join(directory_path, file_name)
if os.path.exists(file_path):
print(f"You are building a {params[0]} {params[1]} {params[2]} ISO type")
process_iso(*params)
else:
print(f"File {file_name} you're not building this ISO type!")
check_build()
2023-07-11 06:48:53 +00:00
def kill_old_iso():
""" Delete older ISOs"""
base_path = BASE_DESTINATION
arch_list = ['deb32/', 'deb64/', 'debarm/', 'dev32/', 'dev64/', 'devarm/']
de_list = ['gnome-flashback', 'openbox', 'xfce']
for archs in arch_list:
for desktops in de_list:
full_path = []
full_path.append(base_path + archs + desktops)
for working_paths in full_path:
N = 1
os.chdir(os.path.join(os.getcwd(), working_paths))
list_of_files = os.listdir()
current_time = time.time()
day = 86400
for fls in list_of_files:
file_location = os.path.join(os.getcwd(), fls)
file_time = os.stat(file_location).st_mtime
if file_time < current_time - day*N:
print(f" Delete : {fls}")
os.remove(file_location)