pyglossary.pyw

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# ui_main.py
#
# Copyright © 2008-2010 Saeed Rasooli <saeed.gnu@gmail.com>    (ilius)
# This file is part of PyGlossary project, https://github.com/ilius/pyglossary
#
# This program is a free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3, or (at your option)
# any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.    See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program. Or on Debian systems, from /usr/share/common-licenses/GPL
# If not, see <http://www.gnu.org/licenses/gpl.txt>.

import os
import sys
import argparse
import builtins
from os.path import dirname, join, realpath
from pprint import pformat
import logging

from pyglossary import core  # essential
from pyglossary import VERSION
from pyglossary.text_utils import startRed, endFormat

# the first thing to do is to set up logger.
# other modules also using logger 'root', so it's essential to set it up prior
# to importing anything else; with exception to pyglossary.core which sets up
# logger class, and so should be done before actually initializing logger.
# verbosity level may be given on command line, so we have to parse arguments
# before setting up logger.
# once more:
# - import system modules like os, sys, argparse etc and pyglossary.core
# - parse args
# - set up logger
# - import submodules
# - other code

# no-progress-bar only for command line UI
# FIXME: load ui-dependent available options from ui modules
# (for example ui_cmd.available_options)
# the only problem is that it has to "import gtk" before it get the
# "ui_gtk.available_options"

# FIXME
# -v (verbose or version?)
# -r (reverse or read-options)

parser = argparse.ArgumentParser(add_help=False)

parser.add_argument(
	'-v',
	'--verbosity',
	action='store',
	dest='verbosity',
	type=int,
	choices=(0, 1, 2, 3, 4),
	required=False,
	default=3,  # FIXME
)
parser.add_argument(
	'--version',
	action='version',
	version='PyGlossary %s' % VERSION,
)
parser.add_argument(
	'-h',
	'--help',
	dest='help',
	action='store_true',
)
parser.add_argument(
	'-u',
	'--ui',
	dest='ui_type',
	default='auto',
	choices=(
		'cmd',
		'gtk',
		'tk',
		# 'qt',
		'auto',
		'none',
	),
)
parser.add_argument(
	'-r',
	'--read-options',
	dest='readOptions',
	default='',
)
parser.add_argument(
	'-w',
	'--write-options',
	dest='writeOptions',
	default='',
)
parser.add_argument(
	'--read-format',
	dest='inputFormat',
)
parser.add_argument(
	'--write-format',
	dest='outputFormat',
	action='store',
)
parser.add_argument(
	'--direct',
	dest='direct',
	action='store_true',
	default=None,
	help='if possible, convert directly without loading into memory',
)
parser.add_argument(
	'--indirect',
	dest='direct',
	action='store_false',
	default=None,
	help='disable `direct` mode, load full data into memory before writing'
		 ', this is default',
)
parser.add_argument(
	'--reverse',
	dest='reverse',
	action='store_true',
)
parser.add_argument(
	'--no-progress-bar',
	dest='progressbar',
	action='store_false',
	default=None,
)
parser.add_argument(
	'--sort',
	dest='sort',
	action='store_true',
	default=None,
)
parser.add_argument(
	'--no-sort',
	dest='sort',
	action='store_false',
	default=None,
)
parser.add_argument(
	'--sort-cache-size',
	dest='sortCacheSize',
	type=int,
	default=None,
)

parser.add_argument(
	'--utf8-check',
	dest='utf8Check',
	action='store_true',
	default=None,
)
parser.add_argument(
	'--no-utf8-check',
	dest='utf8Check',
	action='store_false',
	default=None,
)
parser.add_argument(
	'--lower',
	dest='lower',
	action='store_true',
	default=None,
	help='lowercase words before writing',
)
parser.add_argument(
	'--no-lower',
	dest='lower',
	action='store_false',
	default=None,
	help='don\'t lowercase words before writing',
)
parser.add_argument(
	'--skip-resources',
	dest='skipResources',
	action='store_true',
	default=None,
	help='skip resources (images, audio, etc)',
)

parser.add_argument(
	'--no-color',
	dest='noColor',
	action='store_true',
)
parser.add_argument(
	'inputFilename',
	action='store',
	default='',
	nargs='?',
)
parser.add_argument(
	'outputFilename',
	action='store',
	default='',
	nargs='?',
)

args = parser.parse_args()

log = logging.getLogger('root')
log.setVerbosity(args.verbosity)
log.addHandler(
	core.StdLogHandler(noColor=args.noColor),
)
# with the logger setted up, we can import other pyglossary modules, so they
# can do some loggging in right way.


core.checkCreateConfDir()


##############################

from pyglossary.glossary import Glossary
from ui.ui_cmd import COMMAND, help, parseFormatOptionsStr

##############################


def dashToCamelCase(text):  # converts "hello-PYTHON-user" to "helloPythonUser"
	parts = text.split('-')
	parts[0] = parts[0].lower()
	for i in range(1, len(parts)):
		parts[i] = parts[i].capitalize()
	return ''.join(parts)

ui_list = (
	'gtk',
	'tk',
	'qt',
)

# log.info('PyGlossary %s'%VERSION)


if args.help:
	help()
	sys.exit(0)


if os.sep != '/':
	args.noColor = True


# only used in ui_cmd for now
readOptions = parseFormatOptionsStr(args.readOptions)
writeOptions = parseFormatOptionsStr(args.writeOptions)


"""
	examples for read and write options:
	--read-options testOption=stringValue
	--read-options enableFoo=True
	--read-options fooList=[1,2,3]
	--read-options 'fooList=[1, 2, 3]'
	--read-options 'testOption=stringValue; enableFoo=True; fooList=[1, 2, 3]'
	--read-options 'testOption=stringValue;enableFoo=True;fooList=[1,2,3]'
"""

# FIXME
prefOptionsKeys = (
	# 'verbosity',
	'utf8Check',
	'lower',
	'skipResources',
)

convertOptionsKeys = (
	'direct',
	'progressbar',
	'sort',
	'sortCacheSize',
	# 'sortKey',# or sortAlg FIXME
)

prefOptions = {}
for param in prefOptionsKeys:
	value = getattr(args, param, None)
	if value is not None:
		prefOptions[param] = value

convertOptions = {}
for param in convertOptionsKeys:
	value = getattr(args, param, None)
	if value is not None:
		convertOptions[param] = value

log.pretty(prefOptions, 'prefOptions = ')
log.pretty(readOptions, 'readOptions = ')
log.pretty(writeOptions, 'writeOptions = ')
log.pretty(convertOptions, 'convertOptions = ')

"""
ui_type: User interface type
Possible values:
	cmd - Command line interface, this ui will automatically selected
		  if you give both input and output file
	gtk - GTK interface
	tk - Tkinter interface
	qt - Qt interface
	auto - Use the first available UI
"""
ui_type = args.ui_type


if args.inputFilename:
	if args.outputFilename and ui_type != 'none':
		ui_type = 'cmd'  # silently? FIXME
else:
	if ui_type == 'cmd':
		log.error('no input file given, try --help')
		exit(1)

if ui_type == 'none':
	if args.reverse:
		log.error('--reverse does not work with --ui=none')
		sys.exit(1)
	glos = Glossary()
	glos.convert(
		args.inputFilename,
		inputFormat=args.inputFormat,
		outputFilename=args.outputFilename,
		outputFormat=args.outputFormat,
		readOptions=readOptions,
		writeOptions=writeOptions,
		**convertOptions
	)
	sys.exit(0)
elif ui_type == 'cmd':
	from ui import ui_cmd
	sys.exit(0 if ui_cmd.UI().run(
		args.inputFilename,
		outputFilename=args.outputFilename,
		inputFormat=args.inputFormat,
		outputFormat=args.outputFormat,
		reverse=args.reverse,
		prefOptions=prefOptions,
		readOptions=readOptions,
		writeOptions=writeOptions,
		convertOptions=convertOptions,
	) else 1)
if ui_type == 'auto':
	ui_module = None
	for ui_type2 in ui_list:
		try:
			ui_module = getattr(
				__import__('ui.ui_%s' % ui_type2),
				'ui_%s' % ui_type2,
			)
		except ImportError:
			log.exception('error while importing UI module:')  # FIXME
		else:
			break
	if ui_module is None:
		log.error('no user interface module found!')
		sys.exit(1)
else:
	ui_module = getattr(
		__import__('ui.ui_%s' % ui_type),
		'ui_%s' % ui_type,
	)

sys.exit(0 if ui_module.UI(**prefOptions).run(
	editPath=args.inputFilename,
	readOptions=readOptions,
) else 1)