diff --git a/pyfpdb/SplitHandHistory.py b/pyfpdb/SplitHandHistory.py
index 52c1d340..c996f903 100644
--- a/pyfpdb/SplitHandHistory.py
+++ b/pyfpdb/SplitHandHistory.py
@@ -1,19 +1,22 @@
-#!/usr/bin/python
-# -*- coding: utf-8 -*-
+#!/usr/bin/env python
-#Copyright 2010 Chaz Littlejohn
-#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 .
-#In the "official" distribution you can find the license in agpl-3.0.txt.
+# Copyright 2010, Chaz Littlejohn
+#
+# This program is 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 2 of the License, 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; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+
+########################################################################
import L10n
_ = L10n.get_translation()
@@ -28,6 +31,7 @@ import Options
import Configuration
from Exceptions import *
from cStringIO import StringIO
+import time
(options, argv) = Options.fpdb_options()
@@ -37,7 +41,7 @@ codepage = ["utf-16", "utf-8", "cp1252"]
class SplitHandHistory:
- def __init__(self, config, in_path = '-', out_path = None, hands = 100, filter = "PokerStarsToFpdb", archive = False):
+ def __init__(self, config, in_path = '-', out_path = None, hands = 100, filter = "PokerStarsToFpdb", archive = False, workerid = 0):
self.config = config
self.in_path = in_path
self.out_path = out_path
@@ -50,22 +54,25 @@ class SplitHandHistory:
self.line_addendum = None
self.filedone = False
+ self.timestamp = str(time.time())
+ self.workerid = '%02d' % workerid
+
#Acquire re_SplitHands for this hh
- filter_name = filter.replace("ToFpdb", "")
+ self.filter_name = filter.replace("ToFpdb", "")
mod = __import__(filter)
- obj = getattr(mod, filter_name, None)
+ obj = getattr(mod, self.filter_name, None)
self.re_SplitHands = obj.re_SplitHands
#Determine line delimiter type if any
- if self.re_SplitHands.match('\n\n'):
- self.line_delimiter = '\n\n'
if self.re_SplitHands.match('\n\n\n'):
self.line_delimiter = '\n\n\n'
+ if self.re_SplitHands.match('\n\n'):
+ self.line_delimiter = '\n\n'
#Add new line addendum for sites which match SplitHand to next line as well
- if filter_name == 'OnGame':
+ if self.filter_name == 'OnGame':
self.line_addendum = '*'
- if filter_name == 'Carbon':
+ if self.filter_name == 'Carbon':
self.line_addendum = ' %s' % name
newfile = file(name, 'w')
+ os.chmod(name, 0775)
return newfile
#Archive Hand Splitter
@@ -122,8 +131,11 @@ class SplitHandHistory:
except FpdbEndOfFile:
done = True
break
+ except UnicodeEncodeError:
+ print _('Absurd character done messed you up')
+ sys.exit(2)
except:
- print _("Unexpected error processing file")
+ print _('Unexpected error processing file')
sys.exit(2)
n += 1
outfile.close()
@@ -174,7 +186,7 @@ class SplitHandHistory:
l = infile.readline()
l = l.replace('\r\n', '\n')
outfile.write(l)
- l = infile.readline()
+ l = infile.readline().encode(self.kodec)
while len(l) < 3:
l = infile.readline()
@@ -182,7 +194,7 @@ class SplitHandHistory:
while len(l) > 2:
l = l.replace('\r\n', '\n')
outfile.write(l)
- l = infile.readline()
+ l = infile.readline().encode(self.kodec)
outfile.write(self.line_delimiter)
return infile
@@ -195,13 +207,19 @@ class SplitHandHistory:
def main(argv=None):
if argv is None:
argv = sys.argv[1:]
+
+ if not options.filename:
+ options.filename = sys.argv[1]
if not options.config:
- options.config = Configuration.Config(file = "HUD_config.test.xml")
-
+ options.config = sys.argv[2]
+
+ if sys.argv[3] == "True":
+ options.archive = True
+
if options.filename:
SplitHH = SplitHandHistory(options.config, options.filename, options.outpath, options.hands,
- options.hhc, options.archive)
+ options.hhc, options.archive, options.workerid)
if __name__ == '__main__':
sys.exit(main())