129 lines
4.7 KiB
Python
129 lines
4.7 KiB
Python
from PyQt4 import QtGui, QtCore
|
|
from datetime import *
|
|
import re
|
|
|
|
from generic import PesterIcon
|
|
|
|
class Mood(object):
|
|
moods = ["chummy", "rancorous", "offline", "pleasant", "distraught",
|
|
"unruly", "smooth", "ecstatic", "relaxed", "discontent",
|
|
"devious", "sleek", "detestful", "mirthful", "manipulative",
|
|
"vigorous", "perky", "acceptant", "protective"]
|
|
def __init__(self, mood):
|
|
if type(mood) is int:
|
|
self.mood = mood
|
|
else:
|
|
self.mood = self.moods.index(mood)
|
|
def value(self):
|
|
return self.mood
|
|
def name(self):
|
|
try:
|
|
name = self.moods[self.mood]
|
|
except IndexError:
|
|
name = "chummy"
|
|
return name
|
|
def icon(self, theme):
|
|
|
|
try:
|
|
f = theme["main/chums/moods"][self.name()]["icon"]
|
|
except KeyError:
|
|
return PesterIcon(theme["main/chums/moods/chummy/icon"])
|
|
return PesterIcon(f)
|
|
|
|
class pesterQuirk(object):
|
|
def __init__(self, quirk):
|
|
if type(quirk) != dict:
|
|
raise ValueError("Quirks must be given a dictionary")
|
|
self.quirk = quirk
|
|
self.type = self.quirk["type"]
|
|
def apply(self, string):
|
|
if self.type == "prefix":
|
|
return self.quirk["value"] + string
|
|
if self.type == "suffix":
|
|
return string + self.quirk["value"]
|
|
if self.type == "replace":
|
|
return string.replace(self.quirk["from"], self.quirk["to"])
|
|
if self.type == "regexp":
|
|
return re.sub(self.quirk["from"], self.quirk["to"], string)
|
|
def __str__(self):
|
|
if self.type == "prefix":
|
|
return "BEGIN WITH: %s" % (self.quirk["value"])
|
|
elif self.type == "suffix":
|
|
return "END WITH: %s" % (self.quirk["value"])
|
|
elif self.type == "replace":
|
|
return "REPLACE %s WITH %s" % (self.quirk["from"], self.quirk["to"])
|
|
elif self.type == "regexp":
|
|
return "REGEXP: %s REPLACED WITH %s" % (self.quirk["from"], self.quirk["to"])
|
|
|
|
class pesterQuirks(object):
|
|
def __init__(self, quirklist):
|
|
self.quirklist = []
|
|
for q in quirklist:
|
|
if type(q) == dict:
|
|
self.quirklist.append(pesterQuirk(q))
|
|
elif type(q) == pesterQuirk:
|
|
self.quirklist.append(q)
|
|
def plainList(self):
|
|
return [q.quirk for q in self.quirklist]
|
|
def apply(self, string):
|
|
presuffix = [q for q in self.quirklist if
|
|
q.type=='prefix' or q.type=='suffix']
|
|
replace = [q for q in self.quirklist if
|
|
q.type=='replace' or q.type=='regexp']
|
|
for r in replace:
|
|
string = r.apply(string)
|
|
for ps in presuffix:
|
|
string = ps.apply(string)
|
|
return string
|
|
|
|
def __iter__(self):
|
|
for q in self.quirklist:
|
|
yield q
|
|
|
|
class PesterProfile(object):
|
|
def __init__(self, handle, color=None, mood=Mood("offline"), chumdb=None):
|
|
self.handle = handle
|
|
if color is None:
|
|
if chumdb:
|
|
color = chumdb.getColor(handle, QtGui.QColor("black"))
|
|
else:
|
|
color = QtGui.QColor("black")
|
|
self.color = color
|
|
self.mood = mood
|
|
def initials(self):
|
|
handle = self.handle
|
|
caps = [l for l in handle if l.isupper()]
|
|
if not caps:
|
|
caps = [""]
|
|
return (handle[0]+caps[0]).upper()
|
|
def colorhtml(self):
|
|
return self.color.name()
|
|
def colorcmd(self):
|
|
(r, g, b, a) = self.color.getRgb()
|
|
return "%d,%d,%d" % (r,g,b)
|
|
def plaindict(self):
|
|
return (self.handle, {"handle": self.handle,
|
|
"mood": self.mood.name(),
|
|
"color": unicode(self.color.name())})
|
|
def blocked(self, config):
|
|
return self.handle in config.getBlocklist()
|
|
|
|
def memsg(self, syscolor, suffix, msg):
|
|
uppersuffix = suffix.upper()
|
|
return "<c=%s>-- %s%s <c=%s>[%s%s]</c> %s --</c>" % (syscolor.name(), self.handle, suffix, self.colorhtml(), self.initials(), uppersuffix, msg)
|
|
def pestermsg(self, otherchum, syscolor, verb):
|
|
return "<c=%s>-- %s <c=%s>[%s]</c> %s %s <c=%s>[%s]</c> at %s --</c>" % (syscolor.name(), self.handle, self.colorhtml(), self.initials(), verb, otherchum.handle, otherchum.colorhtml(), otherchum.initials(), datetime.now().strftime("%H:%M"))
|
|
|
|
@staticmethod
|
|
def checkLength(handle):
|
|
return len(handle) <= 256
|
|
@staticmethod
|
|
def checkValid(handle):
|
|
caps = [l for l in handle if l.isupper()]
|
|
if len(caps) != 1 or handle[0].isupper():
|
|
return False
|
|
if re.search("[^A-Za-z0-9]", handle) is not None:
|
|
return False
|
|
return True
|
|
|