From a1c73f6c305c7f74987d99855ba59d8fa823c253 Mon Sep 17 00:00:00 2001 From: Barry Warsaw Date: Wed, 27 Feb 2008 01:26:18 -0500 Subject: Bite the bullet: rename the Mailman package to mailman. --- mailman/database/types.py | 57 +++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 57 insertions(+) create mode 100644 mailman/database/types.py (limited to 'mailman/database/types.py') diff --git a/mailman/database/types.py b/mailman/database/types.py new file mode 100644 index 000000000..acb586849 --- /dev/null +++ b/mailman/database/types.py @@ -0,0 +1,57 @@ +# Copyright (C) 2007-2008 by the Free Software Foundation, Inc. +# +# 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 2 +# 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, write to the Free Software +# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, +# USA. + +__all__ = [ + 'Enum', + ] + + +import sys + +from storm.properties import SimpleProperty +from storm.variables import UnicodeVariable, Variable + + + +class _EnumVariable(Variable): + """Storm variable.""" + + def parse_set(self, value, from_db): + if value is None: + return None + if not from_db: + return value + path, intvalue = value.rsplit(':', 1) + modulename, classname = path.rsplit('.', 1) + __import__(modulename) + cls = getattr(sys.modules[modulename], classname) + return cls[int(intvalue)] + + def parse_get(self, value, to_db): + if value is None: + return None + if not to_db: + return value + return '%s.%s:%d' % (value.enumclass.__module__, + value.enumclass.__name__, + int(value)) + + +class Enum(SimpleProperty): + """Custom munepy.Enum type for Storm.""" + + variable_class = _EnumVariable -- cgit v1.2.3-70-g09d2