fpdb/pyfpdb/Options.py

#!/usr/bin/python
# -*- coding: utf-8 -*-

#Copyright 2008-2010 Ray E. Barker
#This program is free software: you can redistribute it and/or modify
#it under the terms of the GNU Affero General Public License as published by
#the Free Software Foundation, version 3 of the License.
#
#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 Affero General Public License
#along with this program. If not, see <http://www.gnu.org/licenses/>.
#In the "official" distribution you can find the license in agpl-3.0.txt.

import L10n
_ = L10n.get_translation()

import sys
from optparse import OptionParser
#   http://docs.python.org/library/optparse.html

def fpdb_options():

    """Process command line options for fpdb and HUD_main."""
    parser = OptionParser()
    parser.add_option("-x", "--errorsToConsole",
                      action="store_true",
                      help=_("If passed error output will go to the console rather than ."))
    parser.add_option("-d", "--databaseName",
                      dest="dbname",
                      help=_("Overrides the default database name"))
    parser.add_option("-c", "--configFile",
                      dest="config", default=None,
                      help=_("Specifies a configuration file."))
    parser.add_option("-r", "--rerunPython",
                      action="store_true",
                      help=_("Indicates program was restarted with a different path (only allowed once)."))
    parser.add_option("-k", "--konverter",
                      dest="hhc", default="PokerStarsToFpdb",
                      help=_("Module name for Hand History Converter"))
    parser.add_option("-s", "--sitename",
                      dest="sitename", default=None,
                      help=_("A sitename"))
    parser.add_option("-l", "--logging",
                      dest = "log_level", 
                      choices = ('DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL', 'EMPTY'),
                      help = _("Error logging level:")+" (DEBUG, INFO, WARNING, ERROR, CRITICAL, EMPTY)",
                      default = 'EMPTY')
    parser.add_option("-v", "--version", action = "store_true", 
                      help = _("Print version information and exit."))
    parser.add_option("-u", "--usage", action="store_true", dest="usage", default=False,
                    help=_("Print some useful one liners"))
    # The following options are used for SplitHandHistory.py
    parser.add_option("-f", "--file", dest="filename", metavar="FILE", default=None,
                    help=_("Input file"))
    parser.add_option("-D", "--directory", dest="directory", metavar="FILE", default=None,
                    help=_("Input directory"))
    parser.add_option("-o", "--outpath", dest="outpath", metavar="FILE", default=None,
                    help=_("Input out path in quiet mode"))
    parser.add_option("-a", "--archive", action="store_true", dest="archive", default=False,
                    help=_("File to be split is a PokerStars or Full Tilt Poker archive file"))
    parser.add_option("-n", "--numhands", dest="hands", default="100", type="int",
                    help=_("How many hands do you want saved to each file. Default is 100"))
    parser.add_option("--xloc", dest="xloc", default=None, type="int",
                      help=_("X location to open window"))
    parser.add_option("--yloc", dest="yloc", default=None, type="int",
                      help=_("Y location to open Window"))
    parser.add_option("--autoimport", action="store_true", dest="autoimport",
                      help=_("Auto-start Auto-import"))
    parser.add_option("--minimized", action="store_true", dest="minimized",
                      help=_("Start Minimized"))
    parser.add_option("--hidden", action="store_true", dest="hidden",
                      help=_("Start Hidden"))


    (options, argv) = parser.parse_args()
    return (options, argv)

def site_alias(alias):
    """Function for converting various site aliases to the FPDB name"""
    tmp = alias
    aliases = {
                "PokerStars"     : "PokerStars",
                "Full Tilt Poker": "Full Tilt Poker",
                "PartyPoker"     : "PartyPoker",
                "Betfair"        : "Betfair",
                "OnGame"         : "OnGame",
                "Absolute"       : "Absolute",
                "UltimateBet"    : "UltimateBet",
                "Everleaf"       : "Everleaf",
                "Carbon"         : "Carbon",
                "iPoker"         : "iPoker",
                "Winamax"        : "Winamax",
                "Win2day"        : "Win2day",
                "Everest"        : "Everest",
                "Stars"          : "PokerStars",
                "FTP"            : "Full Tilt Poker",
                "Party"          : "PartyPoker",
                "AP"             : "Absolute",
                "UB"             : "UltimateBet",
              }
    try:
        tmp = aliases[alias]
    except KeyError, e:
        tmp = False
        print _("Alias '%s' unknown" % alias)

    return tmp

if __name__== "__main__":
    (options, argv) = fpdb_options()
    print "errorsToConsole =", options.errorsToConsole
    print "database name   =", options.dbname
    print "config file     =", options.config

    print _("press enter to end")
    sys.stdin.readline()