#!/usr/bin/python
+# This is part of ypp-sc-tools, a set of third-party tools for assisting
+# players of Yohoho Puzzle Pirates.
+#
+# Copyright (C) 2009 Ian Jackson <ijackson@chiark.greenend.org.uk>
+#
+# 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 3 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, see <http://www.gnu.org/licenses/>.
+#
+# Yohoho and Puzzle Pirates are probably trademarks of Three Rings and
+# are used without permission. This program is not endorsed or
+# sponsored by Three Rings.
+
+copyright_info = '''
+yoweb-scrape is part of ypp-sc-tools Copyright (C) 2009 Ian Jackson
+This program comes with ABSOLUTELY NO WARRANTY; this is free software,
+and you are welcome to redistribute it under certain conditions.
+For details, read the top of the yoweb-scrape file.
+'''
#---------- setup ----------
import re as regexp
import random
import curses
+import termios
from optparse import OptionParser
+from StringIO import StringIO
from BeautifulSoup import BeautifulSoup
'/Drinking/Spades/Hearts/Treasure Drop/Poker/Distilling'+
'/Alchemistry/Shipwrightery/Blacksmithing/Foraging').split('/')
-standingvals = ('Able/Distinguished/Respected/Master'+
+core_duty_puzzles = [
+ 'Gunning',
+ ['Sailing','Rigging'],
+ 'Bilging',
+ 'Carpentry',
+ ]
+
+duty_puzzles = ([ 'Navigating', 'Battle Navigation' ] +
+ core_duty_puzzles +
+ [ 'Treasure Haul' ])
+
+standingvals = ('Able/Proficient/Distinguished/Respected/Master'+
'/Renowned/Grand-Master/Legendary/Ultimate').split('/')
+standing_limit = len(standingvals)
pirate_ref_re = regexp.compile('^/yoweb/pirate\\.wm')
if opts.debug > 0:
print >>opts.debug_file, m
-def sleep(seconds):
+def debug_flush():
if opts.debug > 0:
- opts.debug_file.flush()
+ opts.debug_file.flush()
+
+def sleep(seconds):
+ debug_flush()
time.sleep(seconds)
def format_time_interval(ti):
skl.msg('puzzle "%s" no standing found' % puzzle)
continue
standing = sl[0]
- for i in range(0, len(standingvals)):
+ for i in range(0, standing_limit):
if standing == standingvals[i]:
self.standings[puzzle] = i
if not puzzle in self.standings:
#---------- pretty-printer for tables of pirate puzzle standings ----------
class StandingsTable:
- def __init__(self, use_puzzles=None, col_width=6, gap_every=5):
+ def __init__(self, f, use_puzzles=None, col_width=6, gap_every=5):
if use_puzzles is None:
if opts.ship_duty:
- use_puzzles=[
- 'Navigating','Battle Navigation',
- 'Gunning',
- ['Sailing','Rigging'],
- 'Bilging',
- 'Carpentry',
- 'Treasure Haul'
- ]
+ use_puzzles=duty_puzzles
else:
use_puzzles=puzzles
self._puzzles = use_puzzles
- self.s = ''
+ self.f = f
self._cw = col_width-1
self._gap_every = gap_every
self._linecount = 0
+ self._o = f.write
+
+ def _nl(self): self._o('\n')
def _pline(self, pirate, puzstrs, extra):
if (self._linecount > 0
and self._gap_every is not None
and not (self._linecount % self._gap_every)):
- self.s += '\n'
- self.s += ' %-*s' % (max(max_pirate_namelen, 14), pirate)
+ self._nl()
+ self._o(' %-*s' % (max(max_pirate_namelen, 14), pirate))
for v in puzstrs:
- self.s += ' %-*.*s' % (self._cw,self._cw, v)
+ self._o(' %-*.*s' % (self._cw,self._cw, v))
if extra:
- self.s += ' ' + extra
- self.s += '\n'
+ self._o(' ' + extra)
+ self._nl()
self._linecount += 1
def _puzstr(self, pi, puzzle):
s += '+' * (standing % 2)
return s
- def headings(self):
+ def headings(self, lhs='', rhs=None):
def puzn_redact(name):
if isinstance(name,list):
return '/'.join(
if spc < 0: return name
return name[0:min(4,spc)] + name[spc+1:]
self._linecount = -2
- self._pline('', map(puzn_redact, self._puzzles), None)
+ self._pline(lhs, map(puzn_redact, self._puzzles), rhs)
self._linecount = 0
def literalline(self, line):
- self.s += line + '\n'
+ self._o(line)
+ self._nl()
self._linecount = 0
def pirate_dummy(self, name, standingstring, extra=None):
self._pline(name, standingstring * len(self._puzzles), extra)
puzstrs = [self._puzstr(pi,puz) for puz in self._puzzles]
self._pline(pi.name, puzstrs, extra)
- def results(self):
- return self.s
#---------- chat log parser ----------
# sorted by pirate name
# you can pass this None and you'll get []
# or True for the current vessel (which is the default)
+ # the returned value is a fresh list of persistent
+ # PirateAboard objects
if vesselname is True: v = self._v
else: v = self._vl.get(vesselname.title())
if v is None: return []
def do_standings_crew_of(args, bu):
ci = prep_crew_of(args, bu, 60)
- tab = StandingsTable()
+ tab = StandingsTable(sys.stdout)
tab.headings()
for (rank, members) in ci.crew:
if not members: continue
for p in members:
pi = PirateInfo(p, random.randint(900,1800))
tab.pirate(pi)
- print tab.results()
class ProgressPrintPercentage:
def __init__(self, f=sys.stdout):
if opts.ship_duty is None: opts.ship_duty = True
displayer = globals()['Display_'+opts.display]()
- rotate_nya = '/-\\'
(myself, track) = prep_chat_log(args, bu, progress=displayer)
- def timeevent(t,e):
- if t is None: return ' ' * 22
- return " %-4s %-16s" % (format_time_interval(now - t),e)
-
displayer.realstart()
+ if os.isatty(0): kr_create = KeystrokeReader
+ else: kr_create = DummyKeystrokeReader
+
+ try:
+ kreader = kr_create(0, 10)
+ ship_aid_core(myself, track, displayer, kreader)
+ finally:
+ kreader.stop()
+ print '\n'
+
+class KeyBasedSorter:
+ def compar_key_pa(self, pa):
+ pi = pa.pirate_info()
+ if pi is None: return None
+ return self.compar_key(pi)
+ def lsort_pa(self, l):
+ l.sort(key = self.compar_key_pa)
+
+class NameSorter(KeyBasedSorter):
+ def compar_key(self, pi): return pi.name
+ def desc(self): return 'name'
+
+class SkillSorter(NameSorter):
+ def __init__(self, relevant):
+ self._want = frozenset(relevant.split('/'))
+ self._avoid = set()
+ for p in core_duty_puzzles:
+ if isinstance(p,basestring): self._avoid.add(p)
+ else: self._avoid |= set(p)
+ self._avoid -= self._want
+ self._desc = '%s' % relevant
+
+ def desc(self): return self._desc
+
+ def compar_key(self, pi):
+ best_want = max([
+ pi.standings.get(puz,-1)
+ for puz in self._want
+ ])
+ best_avoid = [
+ -pi.standings.get(puz,standing_limit)
+ for puz in self._avoid
+ ]
+ best_avoid.sort()
+ def negate(x): return -x
+ debug('compar_key %s bw=%s ba=%s' % (pi.name, `best_want`,
+ `best_avoid`))
+ return (-best_want, map(negate, best_avoid), pi.name)
+
+def ship_aid_core(myself, track, displayer, kreader):
+
def find_vessel():
vn = track.vesselname()
if vn: return (vn, " on board the %s" % vn)
if vn: return (vn, " ashore from the %s" % vn)
return (None, " not on a vessel")
+ def timeevent(t,e):
+ if t is None: return ' ' * 22
+ return " %-4s %-16s" % (format_time_interval(now - t),e)
+
displayer.show(track.myname() + find_vessel()[1] + '...')
+ rotate_nya = '/-\\'
+
+ sort = NameSorter()
+
while True:
track.catchup()
now = time.time()
(vn, s) = find_vessel()
s = track.myname() + s
- s += " at %s\n" % time.strftime("%Y-%m-%d %H:%M:%S")
-
- tbl = StandingsTable()
- tbl.headings()
+ s += " at %s" % time.strftime("%Y-%m-%d %H:%M:%S")
+ s += kreader.info()
+ s += '\n'
aboard = track.aboard(vn)
+ sort.lsort_pa(aboard)
+
+ tbl_s = StringIO()
+ tbl = StandingsTable(tbl_s)
+
+ if track.vesselname(): howmany = ' %d aboard' % len(aboard)
+ else: howmany = ''
+
+ tbl.headings(howmany, ' sorted by '+sort.desc())
for pa in aboard:
pi = pa.pirate_info()
else:
tbl.pirate(pi, xs)
- s += tbl.results()
-
+ s += tbl_s.getvalue()
displayer.show(s)
- sleep(1)
- rotate_nya = rotate_nya[1:3] + rotate_nya[0]
+ tbl_s.close()
+
+ k = kreader.getch()
+ if k is None:
+ rotate_nya = rotate_nya[1:3] + rotate_nya[0]
+ continue
+
+ if k == 'q': break
+ elif k == 'g': sort = SkillSorter('Gunning')
+ elif k == 'c': sort = SkillSorter('Carpentry')
+ elif k == 's': sort = SkillSorter('Sailing/Rigging')
+ elif k == 'b': sort = SkillSorter('Bilging')
+ elif k == 'n': sort = SkillSorter('Navigating')
+ elif k == 'd': sort = SkillSorter('Battle Navigation')
+ elif k == 't': sort = SkillSorter('Treasure Haul')
+ elif k == 'a': sort = NameSorter()
+ else: pass # unknown key command
+
+#---------- individual keystroke input ----------
+
+class DummyKeystrokeReader:
+ def __init__(self,fd,timeout_dummy): pass
+ def stop(self): pass
+ def getch(self): sleep(1); return None
+ def info(self): return ' [noninteractive]'
+
+class KeystrokeReader(DummyKeystrokeReader):
+ def __init__(self, fd, timeout_decisec=0):
+ self._fd = fd
+ self._saved = termios.tcgetattr(fd)
+ a = termios.tcgetattr(fd)
+ a[3] &= ~(termios.ECHO | termios.ECHONL |
+ termios.ICANON | termios.IEXTEN)
+ a[6][termios.VMIN] = 0
+ a[6][termios.VTIME] = timeout_decisec
+ termios.tcsetattr(fd, termios.TCSANOW, a)
+ def stop(self):
+ termios.tcsetattr(self._fd, termios.TCSANOW, self._saved)
+ def getch(self):
+ debug_flush()
+ byte = os.read(self._fd, 1)
+ if not len(byte): return None
+ return byte
+ def info(self):
+ return ''
#---------- main program ----------
display modes (for --display) apply to ship-aid:
--display=dumb just print new information, scrolling the screen
- --display=overwrite use cursor motion, selective clear, etc. to redraw at top
-''')
+ --display=overwrite use cursor motion, selective clear, etc. to redraw at top''')
ao = pa.add_option
ao('-O','--ocean',dest='ocean', metavar='OCEAN', default=None,
help='select ocean OCEAN')
random.seed()
if len(args) < 1:
+ print >>sys.stderr, copyright_info
pa.error('need a mode argument')
if opts.debug_fd is not None: