Config.py

from __future__ import with_statement

import cPickle
from QtUtil import *
from PyQt4.QtCore import *
from PyQt4.QtGui import *
import getpass

wordCache = dict()

try:
    _dbname = getpass.getuser() or "typer"
    if '.' not in _dbname:
        _dbname += '.db'
except:
    _dbname = "typer.db"

class AmphSettings(QSettings):

    defaults = {
            "typer_font": str(QFont("Arial", 22).toString()),
            "history": 30.0 * 24.0,
            "min_chars": 100,
            "max_chars": 200,
            "lesson_stats": 0, # show text/lesson in perf -- not used anymore
            "perf_group_by": 0,
            "perf_items": 100,
            "text_regex": r".*",
            "db_name": _dbname,
            "select_method": 1,
            "num_rand": 50,
            "graph_what": 3,
            "req_space": True,
            "show_last": True,
            "show_xaxis": False,
            "chrono_x": False,
            "dampen_graph": False,

            "minutes_in_sitting": 60.0,
            "dampen_average": 10,
            "def_group_by": 10,

            "use_lesson_stats": False,
            "auto_review": True,

            "repeat" :False,
            "symbols" :False,
            "title_case" : False,
            "stop_symbols" : r""",.?!-'":""",
            "include_symbols" : r"""<0>||0;||"0"||'0'||0||0||0||0;""",
            "symbol_clean" : True,
            "sentence_regex" :r"""\n""",
            "sentence_strip" : r"",
            "phrase_lessons" : True,
            "permissive_errors" : False,
            "invisible_mode" : False,
            "show_repeat" : True,
            "show_since_fail_counter" : True,


            "min_wpm": 0.0,
            "min_acc": 97.0,
            "min_lesson_wpm": 0.0,
            "min_lesson_acc": 100.0,

            "quiz_right_fg": "#000000",
            "quiz_right_bg": "#ffffff",
            "quiz_wrong_fg": "#ffffff",
            "quiz_wrong_bg": "#000000",

            "group_month": 365.0,
            "group_week": 30.0,
            "group_day": 7.0,

            "ana_which": "wpm asc",
            "ana_what": 0,
            "ana_many": 30,
            "ana_source": -1,
            "ana_count": 1,

            "gen_copies": 3,
            "gen_take": 1,
            "gen_mix": 'c',
            #"gen_stats": False,
            "str_clear": 's',
            "str_extra": 10,
            "str_what": 'e'
        }

    def __init__(self, *args):
        super(AmphSettings, self).__init__(QSettings.IniFormat, QSettings.UserScope, "Amphetype", "Amphetype")

    def get(self, k):
        v = self.value(k)
        if not v.isValid():
            return self.defaults[k]
        return cPickle.loads(str(v.toString()))

    def getFont(self, k):
        qf = QFont()
        qf.fromString(self.get(k))
        return qf

    def getColor(self, k):
        return QColor(self.get(k))

    def set(self, k, v):
        p = self.get(k)
        if p == v:
            return
        self.setValue(k, QVariant(cPickle.dumps(v)))
        self.emit(SIGNAL("change"))
        self.emit(SIGNAL("change_" + k), v)

Settings = AmphSettings()

class SettingsColor(AmphButton):
    def __init__(self, key, text):
        self.key_ = key
        super(SettingsColor, self).__init__(Settings.get(key), self.pickColor)
        self.updateIcon()

    def pickColor(self):
        color = QColorDialog.getColor(Settings.getColor(self.key_), self)
        if not color.isValid():
            return
        Settings.set(self.key_, unicode(color.name()))
        self.updateIcon()

    def updateIcon(self):
        pix = QPixmap(32, 32)
        c = Settings.getColor(self.key_)
        pix.fill(c)
        self.setText(Settings.get(self.key_))
        self.setIcon(QIcon(pix))

class SettingsEdit(AmphEdit):
    def __init__(self, setting):
        val = Settings.get(setting)
        typ = type(val)
        validator = None
        if isinstance(val, float):
            validator = QDoubleValidator
        elif isinstance(val, (int, long)):
            validator = QIntValidator
        if validator is None:
            self.fmt = lambda x: x
        else:
            self.fmt = lambda x: "%g" % x
        super(SettingsEdit, self).__init__(
                            self.fmt(val),
                            lambda: Settings.set(setting, typ(self.text())),
                            validator=validator)
        self.connect(Settings, SIGNAL("change_" + setting), self.setText1)

    def setText1(self, x):
        wordCache.clear() # This is dependant on settings which may have just changed
        self.setText(self.fmt(x))


class SettingsCombo(QComboBox):
    def __init__(self, setting, lst, *args):
        super(SettingsCombo, self).__init__(*args)

        prev = Settings.get(setting)
        self.idx2item = []
        for i in range(len(lst)):
            if isinstance(lst[i], basestring):
                # not a tuple, use index as key
                k, v = i, lst[i]
            else:
                k, v = lst[i]
            self.addItem(v)
            self.idx2item.append(k)
            if k == prev:
                self.setCurrentIndex(i)

        self.connect(self, SIGNAL("activated(int)"),
                    lambda x: Settings.set(setting, self.idx2item[x]))

class SettingsCheckBox(QCheckBox):
    def __init__(self, setting, *args):
        super(SettingsCheckBox, self).__init__(*args)
        self.setCheckState(Qt.Checked if Settings.get(setting) else Qt.Unchecked)
        self.connect(self, SIGNAL("stateChanged(int)"),
                    lambda x: Settings.set(setting, True if x == Qt.Checked else False))

class PreferenceWidget(QWidget):
    def __init__(self):
        super(PreferenceWidget, self).__init__()

        self.font_lbl = QLabel()

        self.setLayout(AmphBoxLayout([
            ["Typer font is", self.font_lbl, AmphButton("Change...", self.setFont), None],
            [SettingsCheckBox('auto_review', "Automatically review slow and mistyped words after texts."),
                ('<a href="http://code.google.com/p/amphetype/wiki/Settings">(help)</a>\n', 1)],
            SettingsCheckBox('show_last', "Show last result(s) above text in the Typer."),
            SettingsCheckBox('use_lesson_stats', "Save key/trigram/word statistics from generated lessons."),
            [SettingsCheckBox('req_space', "Make SPACE mandatory before each session"),
                ('<a href="http://code.google.com/p/amphetype/wiki/Settings">(help)</a>\n', 1)],
            None,
            SettingsCheckBox('title_case', "Practice Capitals by Capitlizing the first letter of each word"),
            [SettingsCheckBox('symbols', "Practice Symbols by adding them to each word"), None],
            ["(Skip words containing these characters:", SettingsEdit('stop_symbols'), None],
            ["Symbol patterns ( for example \"0\"||0 will cause half the words to be wrapped in double quotes )", SettingsEdit('include_symbols'), ")", None],
            [SettingsCheckBox('symbol_clean', "Clean then Symbol Replace"), None],
            None,
            ["( Import Lessons: Split lessons regex", SettingsEdit("sentence_regex"), "Strip lessons regex", SettingsEdit('sentence_strip'), " )", None],
             SettingsCheckBox('phrase_lessons', "Include 3 word phrases in lessons"),
             SettingsCheckBox('permissive_errors', "Permissive Errors : no backspace required (restart required)"),
             SettingsCheckBox('invisible_mode', "Invisible Typing Mode"),
             SettingsCheckBox('show_repeat', "Show Repeat Checkbox"),
             SettingsCheckBox('show_since_fail_counter', "Count Perfect Reps"),
            [AmphGridLayout([
                ["INPUT COLORS", "Text Color", "Background"],
                ["Correct Input", SettingsColor('quiz_right_fg', "Foreground"),
                        SettingsColor('quiz_right_bg', "Background")],
                ["Wrong Input", SettingsColor('quiz_wrong_fg', "Foreground"),
                        SettingsColor('quiz_wrong_bg', "Background")],
                [1+1j, 1+2j, 2+1j, 2+2j]
            ]), None],
            None,
            ["Try to limit texts and lessons to between", SettingsEdit("min_chars"),
                "and", SettingsEdit("max_chars"), "characters.", None],
            ["When selecting easy/difficult texts, scan a sample of",
                SettingsEdit('num_rand'), "texts.", None],
            ["When grouping by sitting on the Performance tab, consider results more than",
                SettingsEdit('minutes_in_sitting'), "minutes away to be part of a different sitting.", None],
            ["Group by", SettingsEdit('def_group_by'), "results when displaying last scores and showing last results on the Typer tab.", None],
            ["When smoothing out the graph, display a running average of", SettingsEdit('dampen_average'), "values", None]
        ]))

        self.updateFont()

    def setFont(self):
        font, ok = QFontDialog.getFont(Settings.getFont('typer_font'), self)
        Settings.set("typer_font", unicode(font.toString()))
        self.updateFont()

    def updateFont(self):
        self.font_lbl.setText(Settings.get("typer_font"))
        qf = Settings.getFont('typer_font')
        self.font_lbl.setFont(qf)