Евгений (ХрамычЪ) Храмов
0a651d6fa2
* Добавлена начальная возможность переключения уровня логирования * Исправлены мелкие ошибки
73 lines
3.2 KiB
Python
73 lines
3.2 KiB
Python
import os
|
|
import glob
|
|
import re
|
|
|
|
from .log import *
|
|
|
|
def main_clear_db():
|
|
base_dir = os.path.dirname(os.path.realpath(__file__))
|
|
portwine_db_path = os.path.join(base_dir, 'portwine_db')
|
|
os.makedirs(portwine_db_path, exist_ok=True)
|
|
|
|
for filename in glob.glob(os.path.join(portwine_db_path, '*')): # Установка разрешений на файлы
|
|
os.chmod(filename, 0o644)
|
|
|
|
duplicate_finder = {} #Поиск дубликатов в файлах
|
|
for filename in glob.glob(os.path.join(portwine_db_path, '*')):
|
|
with open(filename, 'r') as file:
|
|
lines = file.readlines()
|
|
for line in lines:
|
|
if '.exe' in line and '#' in line:
|
|
line = line.strip()
|
|
if line not in duplicate_finder:
|
|
duplicate_finder[line] = []
|
|
duplicate_finder[line].append(filename)
|
|
|
|
duplicates = {line: files for line, files in duplicate_finder.items() if len(files) > 1}
|
|
if duplicates:
|
|
log.warning("Обнаружены дубликаты в файлах:")
|
|
for line, files in duplicates.items():
|
|
for file in files:
|
|
log.info(f"{file.split('portwine_db/')[1]} содержит дубликат: {line}")
|
|
exit(1)
|
|
|
|
|
|
for ppdb in glob.glob(os.path.join(portwine_db_path, '*')): # Обработка каждого файла
|
|
log.debug(ppdb)
|
|
|
|
with open(ppdb, 'r') as file: # Удаление определённых строк
|
|
lines = file.readlines()
|
|
|
|
lines = [line for line in lines if not line.startswith(('##export', '##add_'))]
|
|
lines = [line for line in lines if not (
|
|
re.search(r'MANGOHUD|FPS_LIMIT|VKBASALT|_RAY_TRACING|_DLSS|PW_GUI_DISABLED_CS|PW_USE_GAMEMODE|'
|
|
r'PW_USE_SYSTEM_VK_LAYERS|PW_DISABLE_COMPOSITING|PW_USE_EAC_AND_BE|PW_USE_OBS_VKCAPTURE|'
|
|
r'GAMESCOPE|PW_GS', line)
|
|
)]
|
|
|
|
if any(re.search(r'PW_USE_DGVOODOO2="0"|PW_DGVOODOO2="0"', line) for line in lines):
|
|
lines = [line for line in lines if not re.search(r'PW_USE_DGVOODOO2|PW_DGV', line)]
|
|
|
|
|
|
lines = [re.sub(r'export PW_WINE_USE=.*', 'export PW_WINE_USE="WINE_LG"', line)
|
|
if 'PW_WINE_USE="WINE_LG' in line else line for line in lines]
|
|
lines = [re.sub(r'export PW_WINE_USE=.*', 'export PW_WINE_USE="PROTON_LG"', line)
|
|
if 'PW_WINE_USE="PROTON_LG' in line else line for line in lines]
|
|
|
|
with open(ppdb, 'w') as file: # Сохранение изменений
|
|
file.writelines(lines)
|
|
|
|
ppdb_base = os.path.basename(ppdb) # Переименование файлов
|
|
if ppdb_base.endswith('.exe.ppdb'):
|
|
new_name = f"{ppdb_base[:-9]}.ppdb"
|
|
os.rename(ppdb, os.path.join(portwine_db_path, new_name))
|
|
elif ppdb_base.endswith('.EXE.ppdb'):
|
|
new_name = f"{ppdb_base[:-9]}.ppdb"
|
|
os.rename(ppdb, os.path.join(portwine_db_path, new_name))
|
|
elif not ppdb_base.endswith('.ppdb'):
|
|
new_name = f"{ppdb_base}.ppdb"
|
|
os.rename(ppdb, os.path.join(portwine_db_path, new_name))
|
|
|
|
log.info("ГОТОВО!")
|
|
exit(0)
|