merge-dicts.py

#!/usr/bin/env python3
# Dependencies:
# python3, pyglossary, mdict-utils, which
# 
# Install all dependencies with:
# pip3 install mdict-utils python-lzo
# pyglossary better to be installed from a local folder with: python setup.py install (better to use my ready pyglossary zip file)
#
# Import the modules
import os
import sys
import subprocess
import re
import readline

def check_command(command):
    try:
        subprocess.check_output([command, '--version'], stderr=subprocess.STDOUT)
    except subprocess.CalledProcessError:
        return False
    else:
        return True

if check_command('python3'):
    if check_command('pyglossary'):
        if check_command('mdict'):
            print("All dependencies are ready!\n")
        else:
            print("ERROR: mdict not found! Run 'pip3 install mdict-utils'!")
            exit(1)
    else:
        print("ERROR: pyglossary not installed! Run 'pip3 install pyglossary'")
        exit(1)
else:
    print("ERROR: python not installed! Download and install from https://www.python.org/downloads")
    exit(1)

# Ask the user for the file names
num_files = int(input("How many .txt files do you want to merge? "))

output_files = []
for i in range(1, num_files+1):
    file_name = input(f"{i} file name: ")
    output_files.append(file_name)

output = input("Output file name: ")
choice = input("Do you want to sort the files (SLOW if the files are big and not needed if .mtxt)? (y/n): ")

# Use the files in the cat command
if choice.lower() == "y":
    # Sort the files
    cat_command = subprocess.Popen(["cat"] + output_files, stdout=subprocess.PIPE)
    sort_command = subprocess.Popen(["sort"], stdin=cat_command.stdout, stdout=subprocess.PIPE)
    with open(output, "w") as file:
        file.write(sort_command.communicate()[0].decode())
elif choice.lower() == "n":
    # Do not sort the files
    cat_command = subprocess.Popen(["cat"] + output_files, stdout=subprocess.PIPE)
    with open(output, "w") as file:
        file.write(cat_command.communicate()[0].decode())
else:
    # Invalid choice
    print("Invalid option. Please enter y or n.")

src = output
choice3 = input("Convert the resulted .mtxt to mdx (only if the output is .mtxt; but if not .mtxt type 'n')? (y/n): ")

if choice3.lower() == "y":
    # Convert to mdx
    with open("description.html", "w") as file:
        file.write(src[:-4])
    with open("title.html", "w") as file:
        file.write(src[:-4])
    subprocess.run(["mdict", "--title", "title.html", "--description", "description.html", "-a", f"{src[:-4]}mtxt", f"{src[:-4]}mdx"])
    print("All done!")
    exit(1)
elif choice3.lower() == "n":
    # Do not convert to mdx
    print("Your file is not .mtxt or dont want to convert it to mdx, or you you want to sort according to language!")
else:
    # Invalid choice
    print("Invalid option. Please enter y or n.")


src = output

answer = input("Do you want to convert the merged file to .mtxt (with optional headwords sorting)? (y/n) ")
if answer.lower() == 'y':  # Use Word Title option
    print("Your file will be converted to .mtxt!")
elif answer.lower() == 'n':  # Do not use Word Title option
    print("Thank you for using my tool!")
    exit(1)
else:  # Invalid choice
    print("Invalid option. Please enter y or n.")

choice1 = input("Do you want to sort headwords before conversion to mtxt (ex. english then arabic headwords)? (y/n): ")

sort_key = ""
if choice1.lower() == 'y':  # Sort the files
    sort_key = input("Enter the sorting key (e.g. en-ar, ar-en, en, ar) or press enter to use the default (en-ar): ")
    if not sort_key:
        sort_key = "en-ar"
    subprocess.run(["pyglossary", src, "{}.mtxt".format(src[:-4]), "--write-format=OctopusMdictSource", "--sort", "--sort-key=:" + sort_key])

    with open("description.html", "w") as file:
        file.write(src[:-4])
    with open("title.html", "w") as file:
        file.write(src[:-4])

    print('Pyglossary sorting and conversion to mtxt done!')
elif choice1.lower() == 'n':  # Do not sort the files
    subprocess.run(["pyglossary", src, "{}.mtxt".format(src[:-4]), "--write-format=OctopusMdictSource"])
    print('Conversion to mtxt done without sorting!')
else:  # Invalid choice
    print("Invalid option. Please enter y or n.")

choice2 = input("Convert the resulted .mtxt file to mdx? (y/n): ")

if choice2.lower() == 'y':  # Convert to mdx
    subprocess.run(["mdict", "--title", "title.html", "--description", "description.html", "-a", "{}.mtxt".format(src[:-4]), "{}mdx".format(src[:-4])])
    print('All done!')
elif choice2.lower() == 'n':  # Do not convert to mdx
    print('All done!')
else:  # Invalid choice
    print("Invalid option. Please enter y or n.")

print('Thank you for using my tool!')