mirror of
https://codeberg.org/kazephil/Little_CAT_Helpers.git
synced 2023-08-25 14:05:45 +02:00
123 lines
4.1 KiB
Python
123 lines
4.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
'''
|
|
Copy memory and glossary files from memory projects to another folder.
|
|
|
|
A script to collect the "project_save.tmx" and "glossary.txt" files of
|
|
OmegaT projects and copy them to a user-selected centralized location
|
|
for ease of reference between projects or to facilitate their import
|
|
into other CAT tools.
|
|
|
|
The script creates subfolders with the original project name into a
|
|
user-selected central folder, and also renames both the memory and glossary
|
|
files to match the name of the project.
|
|
'''
|
|
|
|
###########################################################################
|
|
#
|
|
# Collect OmegaT Project Data
|
|
# ---------------------------
|
|
# Version 0.2, released July 4, 2022
|
|
# Author: Philippe Tourigny
|
|
# License: GPL3+
|
|
# https://www.gnu.org/licenses/gpl-3.0.en.html
|
|
#
|
|
# Proper and more generalized rewrite of a quick script I'd put together
|
|
# when I had to collect the memories and glossaries from all my OmegaT
|
|
# projects so I could import them in the replacement tool my workplace
|
|
# decided to impose on us.
|
|
#
|
|
# TODO:
|
|
# - Take glossaries other than the default 'glossary.txt" into account.
|
|
# - Enable the customization of paths and files to copy.
|
|
# - Allow users to create new directories from the folder selection
|
|
# dialog.
|
|
###########################################################################
|
|
|
|
import shutil
|
|
import tkinter as tk
|
|
from tkinter import filedialog
|
|
from pathlib import Path
|
|
|
|
# Constants
|
|
|
|
# Default folder for Documents.
|
|
# Set to user home if there is no 'Documents' folder
|
|
DOCHOME = Path(Path.home()/'Documents')
|
|
if not DOCHOME.exists():
|
|
DOCHOME = Path.home()
|
|
|
|
PROJECT = 'omegat.project'
|
|
MEMORY = 'omegat/project_save.tmx'
|
|
GLOSSARY = 'glossary/glossary.txt'
|
|
|
|
|
|
def select_folder(path, title):
|
|
'''Present the user with a dialog to choose the folder they want to use.'''
|
|
|
|
rootWin = tk.Tk()
|
|
rootWin.attributes('-topmost', True)
|
|
rootWin.withdraw()
|
|
|
|
return filedialog.askdirectory(initialdir=path, title=title)
|
|
|
|
|
|
def make_project_list(searchpath):
|
|
'''Build a list of all OmegaT projects in the search path,
|
|
excluding duplicate data in the ".repositories" folder of
|
|
team projects'''
|
|
|
|
# In team projects, the '.repositories' subfolder is two levels
|
|
# above the 'omegat.project' file, hence the need for two 'parent'
|
|
# attributes.
|
|
projects = [omt.parent for omt in searchpath.rglob(PROJECT)
|
|
if not omt.parent.parent.stem == '.repositories']
|
|
|
|
return projects
|
|
|
|
|
|
def collate_project_data(projects):
|
|
'''Get the project name and link it with the paths to its
|
|
'project_save.tmx' and 'glossary.txt' files.'''
|
|
|
|
project_data = {}
|
|
for project in projects:
|
|
# Retrieve the project name and the memory and glossary files
|
|
project_name = project.stem
|
|
|
|
project_data[project_name] = (Path(project, MEMORY),
|
|
Path(project, GLOSSARY))
|
|
|
|
return project_data
|
|
|
|
|
|
def copy_project_data(project_data):
|
|
'''Create a subfolder for each project in the destination folder.
|
|
Copy the corresponding memory and glossary files there, and
|
|
rename them to match the project name.'''
|
|
|
|
destination = Path(select_folder(DOCHOME, 'Select destination folder'))
|
|
|
|
for name, data in project_data.items():
|
|
|
|
# Create project subfolder
|
|
project_folder = Path(destination/name)
|
|
project_folder.mkdir(parents=True, exist_ok=True)
|
|
|
|
for data_file in data:
|
|
|
|
# Make sure the file is valid, give it the same name as the
|
|
# project, and keep its original extension.
|
|
if data_file.exists():
|
|
new_file = Path(project_folder, name+data_file.suffix)
|
|
print('Copying '+str(data_file)+' to '+' '+str(new_file))
|
|
shutil.copy(data_file, new_file)
|
|
|
|
if __name__ == '__main__':
|
|
projects_path = Path(select_folder(DOCHOME,
|
|
'Select the folder to search for OmegaT projects'))
|
|
|
|
projects = make_project_list(projects_path)
|
|
project_data = collate_project_data(projects)
|
|
copy_project_data(project_data)
|