region_tree.py

from apps.genapi import models


class RegionTreeManager(models.Manager):

    def handle_qs(self, qs):
        not_detailed_query = "not find_in_set('detailed', Options)"
        not_deleted_query = "not find_in_set('deleted', Options)"
        qs = qs.extra(where=[not_detailed_query, not_deleted_query])
        return qs

    def get_queryset(self):
        qs = super(RegionTreeManager, self).get_queryset()
        return self.handle_qs(qs)

    def get_query_set(self):
        qs = super(RegionTreeManager, self).get_query_set()
        return self.handle_qs(qs)


class RegionTree(models.Handbook):

    objects = RegionTreeManager()

    regionid = models.IntegerField(
        primary_key=True,
        db_column='RegionID'
    )

    regionparentid = models.IntegerField(
        db_column='RegionParentID',
        default=0
    )

    geotype = models.IntegerField(
        db_column='GeoType',
        default=0
    )

    options = models.SetField(
        max_length=36,
        db_column='Options',
        blank=True
    )

    bannerflags = models.IntegerField(
        db_column='BannerFlags',
        default=0
    )

    offset = models.IntegerField(
        db_column='Offset',
        default=0
    )

    class Meta(object):
        db_table = 'RegionTree'
        app_label = 'pybsadm'

    def __str__(self):
        return '%s %s' % (self.regionid, self.regionparentid)

    def __unicode__(self):
        return u'%s %s' % (self.regionid, self.regionparentid)