/usr/lib/python3/dist-packages/sqlalchemy_utils/types/weekdays.py is in python3-sqlalchemy-utils 0.32.21-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 | import six
from sqlalchemy import types
from .. import i18n
from ..exceptions import ImproperlyConfigured
from ..primitives import WeekDay, WeekDays
from .bit import BitType
from .scalar_coercible import ScalarCoercible
class WeekDaysType(types.TypeDecorator, ScalarCoercible):
"""
WeekDaysType offers way of saving WeekDays objects into database. The
WeekDays objects are converted to bit strings on the way in and back to
WeekDays objects on the way out.
In order to use WeekDaysType you need to install Babel_ first.
.. _Babel: http://babel.pocoo.org/
::
from sqlalchemy_utils import WeekDaysType, WeekDays
from babel import Locale
class Schedule(Base):
__tablename__ = 'schedule'
id = sa.Column(sa.Integer, autoincrement=True)
working_days = sa.Column(WeekDaysType)
schedule = Schedule()
schedule.working_days = WeekDays('0001111')
session.add(schedule)
session.commit()
print schedule.working_days # Thursday, Friday, Saturday, Sunday
WeekDaysType also supports scalar coercion:
::
schedule.working_days = '1110000'
schedule.working_days # WeekDays object
"""
impl = BitType(WeekDay.NUM_WEEK_DAYS)
def __init__(self, *args, **kwargs):
if i18n.babel is None:
raise ImproperlyConfigured(
"'babel' package is required to use 'WeekDaysType'"
)
super(WeekDaysType, self).__init__(*args, **kwargs)
@property
def comparator_factory(self):
return self.impl.comparator_factory
def process_bind_param(self, value, dialect):
if isinstance(value, WeekDays):
value = value.as_bit_string()
if dialect.name == 'mysql':
func = bytes if six.PY3 else bytearray
return func(value, 'utf8')
return value
def process_result_value(self, value, dialect):
if value is not None:
return WeekDays(value)
def _coerce(self, value):
if value is not None and not isinstance(value, WeekDays):
return WeekDays(value)
return value
|