#!@PYTHON@
# -*- mode: Python; coding: utf-8; -*-
# PO file spell checker
#
# Pedro Morais <morais@kde.org>
# José Nuno Pires <jncp@netcabo.pt>
# João Neves <joao@silvanaves.org>
# (c) Copyright 2003, 2004
# Distributable under the terms of the GPL - see COPYING
import sys
import os
import getopt
import re
if not "@prefix@/share/@PACKAGE@" in sys.path:
sys.path.append("@prefix@/share/@PACKAGE@")
from POFile import POFile
from util import Output
spellCommand = 'aspell --encoding=UTF-8 list'
def usage(code = -1):
w = sys.stderr.write
w('Usage: POFileSpell [OPTION] <FILE>...\n')
w('\n')
w('Mandatory arguments to long options are mandatory '
'for short options too.\n')
w('\n')
w('Options:\n')
w(' -h, --help show this help\n')
w(' -i, --interactive interactive console mode\n')
w(' -o, --overview generate an overview file\n')
w(' -d, --dict=<file> load a extra dictionary file\n')
w(' --batch-add=<file> add words on file to ignore list\n')
w(' --command=<command> command used for spellchecking\n')
w(' (default: "%s")\n' % spellCommand)
sys.exit(code)
try:
opts, args = getopt.getopt(sys.argv[1:], "hiod:",
["help", "interactive", "overview", "dict=",
"command=", "batch-add="])
except getopt.GetoptError:
usage()
interactive = 0
overview = None
dict = {}
dictfiles = []
batchaddwordlist = []
for o, a in opts:
if o in ("-h", "--help"): usage(0)
if o in ("-i", "--interactive"): interactive = 1
if o in ("-o", "--overview"): overview = {}
if o in ("-d", "--dict"):
fn = os.path.expandvars(os.path.expanduser(a))
dictfiles.append(fn)
words = open(fn).readlines()
for word in words:
word = word[:-1]
dict[word] = word
if o in ("--batch-add",):
fn = os.path.expandvars(os.path.expanduser(a))
words = open(fn).readlines()
for word in words:
word = word[:-1]
batchaddwordlist.append(word)
if o in ("--command",):
spellCommand = a
if len(args) < 1: usage()
def spelling(args, overview, dict, dictfiles, spellCommand):
tagname = "po-file-spell"
if overview != None: tagname = "po-file-spell-overview"
out = Output(tagname)
for filename in args:
po = POFile(filename)
if po.parse() == 0:
sys.stderr.write('error parsing file %s\n' % filename)
else:
po.parseHeader()
if spellCommand: po.spellCommand = spellCommand
po.spell(dict)
if len(po.spellErrors) > 0:
if overview != None:
for e in po.spellErrors:
ce = 0
if overview.has_key(e): ce= overview[e]
overview[e] = ce + 1
else:
out.opentag('file', {'name': filename})
for e in po.spellErrors:
out.opentag('error', {}, body = e, close = 1)
out.closetag()
if overview:
for i in overview.iteritems():
out.opentag('error', {'count': str(i[1])}, body = i[0], close = 1)
out.finish()
def interactiveSpelling(args, dict, dictfiles, spellCommand):
for filename in args:
po = POFile(filename)
if po.parse() == 0:
sys.stderr.write('error parsing file %s\n' % filename)
else:
po.parseHeader()
if spellCommand: po.spellCommand = spellCommand
po.spell(dict)
if len(po.spellErrors) > 0:
for e in po.spellErrors:
print 'Error: %s' % e
ctxs = po.searchInMsgstr(e)
for line, message, context in ctxs:
print ' message %d, line %d: %s' % (
message, line, context)
print 'a - add to current file (%s)' % filename
if len(ctxs):
print 'r - replace in current file with another word'
d = 1
for dictfile in dictfiles:
print '%d - add to dictionary %s' % (d, dictfile)
d = d + 1
print 'n - go to the next file'
x = raw_input('Enter your choice, enter to continue: ')
if x == 'n':
break
elif x == 'a':
if addSpellExtra(po, filename, e):
print 'added word to file %s' % filename
else:
print 'error adding word to file %s' % filename
elif len(ctxs) and x == 'r':
replace = raw_input('Enter replacement word: ')
for line, message, context in ctxs:
podata = po.data[message - 1]
regexp = re.compile('\\b' + e + '\\b')
rrr = regexp.sub(replace, podata[3])
lines = po.prepare_replace(message)
if lines != None:
file = open(filename, 'w')
po.execute_replace(lines, rrr, 1, file,
breaknewlines = 1)
file.close()
po.data[message - 1] = (podata[0], podata[1],
podata[2], rrr,
podata[3])
print 'replaced word in %s' % filename
else:
print 'error replacing word in %s' % filename
else:
dictchoice = 0
try:
dictchoice = int(x)
except:
dictchoice = 0
if dictchoice >= 1 and dictchoice <= len(dictfiles):
chdict = dictfiles[dictchoice - 1]
dfile = open(chdict, 'a')
dfile.write('%s\n' % e)
dfile.close()
print 'added to dictionary %s' % chdict
dict[e] = e
print
def batchAddSpellExtra(args, wordlist):
for filename in args:
po = POFile(filename)
if po.parse() == 0:
sys.stderr.write('error parsing file %s\n' % filename)
return
else:
po.parseHeader()
for word in wordlist:
addSpellExtra(po, filename, word)
def addSpellExtra(po, filename, e):
header = po.data[0][3]
headerlines = header.split('\\n')
li = 0
sli = -1
for hl in headerlines:
if hl.startswith('X-POFile-SpellExtra: '):
if len(hl) + len(e) < 74:
sli = li
else:
sli = -1
li = li + 1
if sli < 0:
header = header + ('X-POFile-SpellExtra: %s\\n' % e)
else:
header = ''
li = 0
for hl in headerlines:
header = header + hl
if li == sli:
header = header + ' ' + e
header = header + '\\n'
li = li + 1
lines = po.prepare_replace(1)
if lines != None:
file = open(filename, 'w')
po.execute_replace(lines, header, 1, file, breaknewlines = 1)
file.close()
po.data[0] = (po.data[0][0], po.data[0][1],
po.data[0][2], header, po.data[0][3])
return 1
return 0
if len(batchaddwordlist):
batchAddSpellExtra(args, batchaddwordlist)
elif interactive:
interactiveSpelling(args, dict, dictfiles, spellCommand)
else:
spelling(args, overview, dict, dictfiles, spellCommand)