summaryrefslogtreecommitdiff
path: root/lib/param
diff options
context:
space:
mode:
Diffstat (limited to 'lib/param')
-rw-r--r--lib/param/__init__.py65
-rw-r--r--lib/param/bool_.py19
-rw-r--r--lib/param/color.py17
-rw-r--r--lib/param/enum.py8
-rw-r--r--lib/param/float_.py19
-rw-r--r--lib/param/img_url.py98
-rw-r--r--lib/param/int_.py18
-rw-r--r--lib/param/json.py8
-rw-r--r--lib/param/raw.py6
-rw-r--r--lib/param/string.py15
10 files changed, 273 insertions, 0 deletions
diff --git a/lib/param/__init__.py b/lib/param/__init__.py
new file mode 100644
index 0000000..bdf5885
--- /dev/null
+++ b/lib/param/__init__.py
@@ -0,0 +1,65 @@
+import time
+import sys
+
+from config import WORKING_DIR
+
+class BadParamError(Exception):
+ pass
+
+
+class Param(object):
+ def __init__(self, classname="", **kwargs):
+ self._working_dir = WORKING_DIR
+ self._now = kwargs.get("now", str(int(time.time())));
+ self._classname = classname
+ for key, value in kwargs.items():
+ setattr(self, key, value)
+
+ def __nonzero__(self):
+ return True if self.value else False
+
+ def __str__(self):
+ return str(self.value)
+
+ def __eq__(self, other):
+ return self.value == other
+
+ def __ne__(self, other):
+ return self.value != other
+
+ def set_val(self, value):
+ try:
+ self.value = value
+ except Exception as e:
+ self.err_warn("Unable to set value {}".format(value))
+
+ def err_warn(self, s, error=None):
+ self._error_log(s, error=error);
+ raise BadParamError("%s - %s" % (self._classname, s))
+
+ def __getattr__(self, key):
+ try:
+ return self.__getattribute__(key);
+ except AttributeError:
+ return None
+
+ def err_fatal(self, s, error=None):
+ self._log(s, error, fatal=True);
+ sys.exit(1);
+
+ def _error_log(self, s, error=None, fatal=False):
+ message = "ERROR - BAD PARAM"
+ if fatal: message += "- [FATAL] -"
+ sys.stderr.write("{}:{} - {}\n".format(message, self._classname, s))
+ if error:
+ sys.stderr.write("PARAM ERROR: {}\n".format(str(error)))
+
+from param.int_ import ParamInt
+from param.raw import ParamRaw
+from param.bool_ import ParamBool
+from param.enum import ParamEnum
+from param.json import ParamJson
+from param.color import ParamColor
+from param.float_ import ParamFloat
+from param.img_url import ParamImg_url
+from param.string import ParamString
diff --git a/lib/param/bool_.py b/lib/param/bool_.py
new file mode 100644
index 0000000..e275115
--- /dev/null
+++ b/lib/param/bool_.py
@@ -0,0 +1,19 @@
+from param import Param
+import re
+import sys
+class ParamBool(Param):
+ def __init__(self, value, classname=""):
+ super(ParamBool, self).__init__(classname=classname)
+ if value:
+ self.value = self._bool_correct(value)
+ else:
+ self.value = False
+ def _bool_correct(self, b):
+ if type(b) == str or type(b) == unicode:
+ if re.match(r'true', b, re.IGNORECASE):
+ return True
+ elif re.match(r'false', b, re.IGNORECASE):
+ return False
+ elif type(b) == bool:
+ return b
+ self.err_warn("Not a bool: %s" % str(b))
diff --git a/lib/param/color.py b/lib/param/color.py
new file mode 100644
index 0000000..5293642
--- /dev/null
+++ b/lib/param/color.py
@@ -0,0 +1,17 @@
+from param import Param
+import re
+class ParamColor(Param):
+ def __init__(self, value, classname=""):
+ super(ParamColor, self).__init__(classname=classname)
+ try:
+ self.value = self._color_sanitize(value)
+ except Exception as e:
+ self.err_warn("Unable to sanitize the color: %s" % str(value))
+ self.err_warn(str(e))
+ def _color_sanitize(self, s):
+ if s == "":
+ return "transparent"
+ if re.match('(rgba?\([0-9]+,[0-9]+,[0-9]+\))|([a-zA-Z]+)|(\#[A-Ha-h0-9]+)', s):
+ return s.replace(' ', '');
+ else:
+ self.err_warn("Not a color: {}\n".format(s))
diff --git a/lib/param/enum.py b/lib/param/enum.py
new file mode 100644
index 0000000..1dd510a
--- /dev/null
+++ b/lib/param/enum.py
@@ -0,0 +1,8 @@
+from param import Param
+import sys
+class ParamEnum(Param):
+ def __init__(self, value, enum_values=[], classname=""):
+ super(ParamEnum,self).__init__(classname=classname)
+ if value not in enum_values:
+ return self.err_warn("Value %s not in enum values" % str(value))
+ self.value = value
diff --git a/lib/param/float_.py b/lib/param/float_.py
new file mode 100644
index 0000000..a67bd44
--- /dev/null
+++ b/lib/param/float_.py
@@ -0,0 +1,19 @@
+from param import Param
+
+class ParamFloat(Param):
+ def __init__(self, value, classname=""):
+ self._classname = classname
+ super(ParamFloat, self).__init__(classname=classname)
+ try:
+ if value:
+ self.value = float(value)
+ else:
+ self.value = 0.0
+ except Exception as e:
+ self.err_warn("Not a float: %s" % str(value))
+ self.err_warn(str(e))
+ def __int__(self):
+ return int(self.value)
+
+ def __float__(self):
+ return float(self.value)
diff --git a/lib/param/img_url.py b/lib/param/img_url.py
new file mode 100644
index 0000000..a73fdd0
--- /dev/null
+++ b/lib/param/img_url.py
@@ -0,0 +1,98 @@
+import os
+from param import Param
+from config import *
+import urllib, urllib2
+from subprocess import Popen, PIPE
+import sys
+Request = urllib2.Request
+urlencode = urllib.urlencode
+urlopen = urllib2.urlopen
+Request = urllib2.Request
+urlencode = urllib.urlencode
+urlopen = urllib2.urlopen
+
+import sys;
+class ParamImg_url(Param):
+ def __init__(self, value, key="", classname=""):
+ super(ParamImg_url, self).__init__(classname=classname)
+ if value:
+ try:
+ self.filename = self._filename_temporary(key)
+
+ self.path = os.path.join(self._working_dir, self.filename)
+ self._image_download(value, self.path)
+ self.mimetype = self._image_mimetype(self.path)
+ self.url = value
+ except Exception as e:
+ self.err_warn("Unable to download image: %s" % str(value))
+ self.err_warn(str(e))
+
+ def _filename_temporary(self, s):
+ return "_tmp-{}-{}_{}".format(self._classname, self._now, s)
+
+ def __dict__(self):
+ return {
+ 'filename' : self.filename,
+ 'path': self.path,
+ 'url': self.url,
+ 'mimetype': self.mimetype
+ }
+
+ def __getitem__(self, item):
+ return self.__dict__().__getitem__(item)
+
+ def __str__(self):
+ return str(self.__dict__())
+
+ def __nonzero__(self):
+ return True if self.path and self.mimetype else False
+
+ def _image_download(self, url, path):
+ max_size = MAX_SIZE
+ if self.username in SPECIAL_DOWNLOADERS:
+ max_size = SPECIAL_DOWNLOADERS_MAX_SIZE
+ try:
+ self._download(url, path, max_size=max_size)
+ except Exception as e:
+ self.err_warn("Download failed");
+
+ def _browser_request (self, url, data=None):
+ headers = {
+ 'User-Agent': 'Mozilla/4.0 (compatible; MSIE 5.5; Windows NT)',
+ 'Accept': '*/*',
+ }
+ try:
+ req = Request(url, data, headers)
+ response = urlopen(req)
+ except IOError as e:
+ if hasattr(e, 'code'):
+ sys.stderr.write( 'browser request error: %s - ERROR %s' % (url, e.code) )
+ raise IOError
+ return response
+
+ def _download(self, url, destination, max_size=MAX_SIZE):
+ response = self._browser_request(url, None)
+
+ rawimg = response.read()
+ if len(rawimg) == 0:
+ self.err_warn("got zero-length file")
+ if len(rawimg) > max_size:
+ self.err_warn("file too big: max size {} KB / {} is {} KB".format(
+ str(MAX_SIZE/1024),
+ destination,
+ str(len(rawimg)/1024)
+ )
+ )
+ f = open(destination, "w")
+ f.write(rawimg)
+ f.close()
+
+ def _image_mimetype(self, f):
+ try:
+ mimetype = Popen(
+ [BIN_IDENTIFY, f], stdout=PIPE
+ ).communicate()[0].split(" ")[1].lower()
+ return mimetype
+ except Exception as e:
+ sys.stderr.write("couldn't determine mimetype\n")
+ raise e;
diff --git a/lib/param/int_.py b/lib/param/int_.py
new file mode 100644
index 0000000..fd9730d
--- /dev/null
+++ b/lib/param/int_.py
@@ -0,0 +1,18 @@
+from param import Param
+
+class ParamInt(Param):
+ def __init__(self, value, classname=""):
+ super(ParamInt, self).__init__(classname=classname)
+ try:
+ if value:
+ self.value = int(value)
+ else:
+ self.value = 0
+ except Exception as e:
+ self.err_warn("Not an int: %s" % str(value))
+ self.err_warn(str(e))
+ def __int__(self):
+ return int(self.value)
+
+ def __float__(self):
+ return float(self.value)
diff --git a/lib/param/json.py b/lib/param/json.py
new file mode 100644
index 0000000..8897710
--- /dev/null
+++ b/lib/param/json.py
@@ -0,0 +1,8 @@
+from param import Param
+import simplejson as json
+
+class ParamJson(Param):
+ def __init__(self, value, classname=""):
+ super(ParamJson, self).__init__(classname=classname)
+ self.value = json.parse(value)
+
diff --git a/lib/param/raw.py b/lib/param/raw.py
new file mode 100644
index 0000000..f9cdea9
--- /dev/null
+++ b/lib/param/raw.py
@@ -0,0 +1,6 @@
+from param import Param
+
+class ParamRaw(Param):
+ def __init__(self, value, classname=""):
+ super(ParamRaw, self).__init__(classname=classname)
+ self.value = value
diff --git a/lib/param/string.py b/lib/param/string.py
new file mode 100644
index 0000000..91a2d34
--- /dev/null
+++ b/lib/param/string.py
@@ -0,0 +1,15 @@
+from param import Param
+import re
+import sys
+class ParamString(Param):
+ def __init__(self, value, classname=""):
+ super(ParamString, self).__init__(classname=classname)
+ if value:
+ try:
+ self.value = self.sanitize(value)
+ except Exception as e:
+ self.err_warn("Unable to sanitize: %s\nreason:%s" % (str(value), str(e)))
+ else:
+ self.value = ""
+ def sanitize (self, s):
+ return re.sub(r'\W+', '', s)
generated by cgit v1.2.3-70-g09d2 (git 2.51.0) at 2025-12-15 13:01:47 +0000