KosLookupExe.py

import ctypes
import ctypes.wintypes
import datetime
import os

import wx

try:
  import winsound
except ImportError:
  winsound = None

import ChatKosLookup


DIVIDER = '-' * 40
PLUS_TAG = '[+]'
MINUS_TAG = u'[\u2212]'  # Unicode MINUS SIGN


# Cargo-culted from:
# http://stackoverflow.com/questions/3927259/how-do-you-get-the-exact-path-to-my-documents
def GetMyDocumentsDir():
  shell32 = ctypes.windll.shell32
  buf = ctypes.create_unicode_buffer(ctypes.wintypes.MAX_PATH + 1)
  if shell32.SHGetSpecialFolderPathW(None, buf, 0x5, False):
    return buf.value
  return None


def GetEveLogsDir():
  home = GetMyDocumentsDir()
  if not home:
    return None
  if os.path.isdir(os.path.join(home, 'EVE', 'logs', 'Chatlogs')):
    return os.path.join(home, 'EVE', 'logs', 'Chatlogs')
  if os.path.isdir(os.path.join(home, 'CCP', 'EVE', 'logs', 'Chatlogs')):
    return os.path.join(home, 'CCP', 'EVE', 'logs', 'Chatlogs')
  return None


class MainFrame(wx.Frame):
  def __init__(self, *args, **kwargs):
    wx.Frame.__init__(self, *args, **kwargs)
    self.UpdateIcon()
    self.working_file = self.GetWorkingFile()
    if not self.working_file:
      self.Close()
      return
    self.UpdateTitle(self.working_file)
    self.checker = ChatKosLookup.KosChecker()
    self.tailer = ChatKosLookup.FileTailer(self.working_file)
    self.labels = []
    self.text_boxes = []
    for i in xrange(100):
      text = wx.StaticText(self, -1, '', (5, 16 * i + 5))
      self.text_boxes.append(text)
    self.SetSize((150, 800))
    self.SetBackgroundColour('white')
    self.Show()
    self.KosCheckerPoll()

  def UpdateIcon(self):
    """
    If running from py2exe, then the icon is implicitly obtained from the .exe
    file, but when running from source, this method pulls it in from the
    directory containing the python modules.
    """
    try:
      icon_path = os.path.join(os.path.dirname(__file__), 'icon.ico')
    except NameError:
      # __file__ does not exist
      return
    if os.path.exists(icon_path):
      self.SetIcon(wx.Icon(icon_path, wx.BITMAP_TYPE_ICO))

  def KosCheckerPoll(self):
    entry, comment = self.tailer.poll()
    if not entry:
      wx.FutureCall(1000, self.KosCheckerPoll)
      return
    kos, not_kos, error = self.checker.koscheck_logentry(entry)
    new_labels = []
    if comment:
      new_labels.append(('black', comment))
    if kos or not_kos:
      new_labels.append(('black',
                        'KOS: %d  Not KOS: %d' % (len(kos), len(not_kos))))
    if kos:
      self.PlayKosAlertSound()
      new_labels.extend([('red', u'%s %s (%s)' % (MINUS_TAG, p, reason))
                         for (p, reason) in kos])
    if not_kos:
      if kos:
        new_labels.append(('black', ' '))
      new_labels.extend([('blue', '%s %s' % (PLUS_TAG, p)) for p in not_kos])
    if error:
      new_labels.append(('black', 'Error: %d' % len(error)))
      new_labels.extend([('black', p) for p in error])
    if new_labels:
      new_labels.append(('black', DIVIDER))
    self.labels = new_labels + self.labels
    self.labels = self.labels[:100]
    self.UpdateLabels()
    wx.FutureCall(100, self.KosCheckerPoll)

  def PlayKosAlertSound(self):
    if winsound:
      winsound.PlaySound("SystemQuestion", winsound.SND_ALIAS)

  def UpdateLabels(self):
    for i, (color, label) in enumerate(self.labels):
      self.text_boxes[i].SetForegroundColour(color)
      self.text_boxes[i].SetLabel(label)

  def UpdateTitle(self, working_file):
    filename = os.path.basename(working_file)
    name = filename.rsplit('_', 2)[0]
    name.replace('_', ' ')
    self.SetLabel(name)

  def GetWorkingFile(self):
    today = datetime.date.today().strftime('%Y%m%d')
    wildcards = [
        'Fleet logs (today)', 'Fleet_%s_*.txt' % today,
        'Fleet logs (all)', 'Fleet*.txt',
        'All logs (today)', '*_%s_*.txt' % today,
        'All logs', '*.txt']
    dialog = wx.FileDialog(
        self,
        'Choose a log file',
        GetEveLogsDir(),
        style=wx.OPEN,
        wildcard='|'.join(wildcards))
    result = dialog.ShowModal()
    if result != wx.ID_OK:
      return None
    return dialog.GetPath()


if __name__ == '__main__':
  app = wx.App()
  frame = MainFrame(None, -1, 'KOS Checker')
  app.MainLoop()