ota-flasher/app/utils.py

33 lines
1 KiB
Python
Raw Permalink Normal View History

2024-06-09 17:38:32 +00:00
import os
import re
import shutil
def count_versions(folder_path):
versions = set()
version_pattern = re.compile(r'^(\d+\.\d+\.\d+)')
for file_name in os.listdir(folder_path):
match = version_pattern.match(file_name)
if match:
versions.add(match.group(1))
return len(versions)
def keep_latest_versions(folder_path, num_versions_to_keep=2):
version_files = {}
version_pattern = re.compile(r'^(\d+\.\d+\.\d+)')
for file_name in os.listdir(folder_path):
match = version_pattern.match(file_name)
if match:
version = match.group(1)
if version not in version_files:
version_files[version] = []
version_files[version].append(file_name)
versions_sorted = sorted(version_files.keys(), reverse=True)
versions_to_remove = versions_sorted[num_versions_to_keep:]
for version in versions_to_remove:
for file_name in version_files[version]:
os.remove(os.path.join(folder_path, file_name))