KosLookupExe.py

import ctypes
import ctypes.wintypes
import datetime
import os

import wx

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.working_file = self.GetWorkingFile()
    if not self.working_file:
      self.Close()
      return
    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 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:
      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 UpdateLabels(self):
    for i, (color, label) in enumerate(self.labels):
      self.text_boxes[i].SetForegroundColour(color)
      self.text_boxes[i].SetLabel(label)

  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()