上一篇,我只是做了。 默认的显示。
def __iter__(self): '''默认显示。 用户可以自定制''' if isinstance(self.queryset_or_tuple, list): for item in self.queryset_or_tuple: yield "<a href='#'>%s</a>" % item[1] else: for item in self.queryset_or_tuple: if isinstance(item, Model): print(item) yield "<a href='#'>%s</a>" % item
能看出的是, 我返回的这个 yield 只是需要一个, 文本的信息。 而且是在这里做了判断,类型。
改进一下, 既然我只需要一个文本信息。 那么 这个工作, 应该交给 使用者,自己进行指定。
比如 默认返回 <a href='#'>男</a> 我想加点东西,比如 <a href='#'>男666</a>
而且,并不是所有的都加。 如果我直接在 __iter__ 中。 给他加666。 也是可以的, 只需要在 search_group 列表。 加入对象的时候。为对象, 添加一个参数就可以了。
但是我想换一种方式, 这个参数还是需要制定的, 但是获取文本的这个工作。 我交给 Option 类来做。 在类中添加一个参数。用于在最初的类型判断时, 就指定出,当前的 这个 field 对象。 是一个 choice 还是一个 Model 对象。
class Option(object): '''使用组合搜索, 想要一些自己的搜索条件。 可扩展可继承''' def __init__(self, field, db_condition=None, text_func=None): ''' :param filed: 组合搜索关联的字段 :param db_condition: 数据库关联查询时查询的条件 :param text_func: 此函数用于,页面组合搜索的文本和样式 ''' self.field = field self.db_condition = db_condition self.text_func = text_func if not db_condition: db_condition = {} self.db_condition = db_condition self.is_choice = False # 用于判断field对象里面是一个 choice列表 or 外键的queryset # 添加的函数就是这个, def get_text_func(self, field_obj): ''' 获取文本,的函数。(从) :param field_obj: (1, "男") or model对象 :return: ''' if self.text_func: # 这个参数就是让用户来自己制定的。 下面展示如何使用。 return self.text_func(field_obj)
# 通过 is_choice 判断是否是 choice。 决定返回的是,元组中的 第二个元素。 还是关联表的表名(verbose_name="部门") if self.is_choice: return field_obj[1] return str(field_obj) def get_db_condition(self, request, *args, **kwargs): '''预留继承后的重写函数, 重写后,次基类中的该方法,将被覆盖。 默认返回的是开发者输入的值。''' '''重写后, 可根据,前端的返回值,进行一定的判断''' return self.db_condition def get_queryset_or_tuple(self, model_class, request, *args, **kwargs): '''根据字段去获取数据库关联的数据''' # 根据gender或者classes字符串,组自己对应的model类中,找到字段对象,再根据对象,获取关联的数据 field_obj = model_class._meta.get_field(self.field) # 固定用法mate 类,中的get_field() 就可以根据字符串获取对应的对象 # 对field_obj 的类型做判断。 来确定他是一个 choice 还是一个 foreignkey 外键 if isinstance(field_obj, ForeignKey) or isinstance(field_obj, ManyToManyField): # 获取关联表中的, 数据 django1.0 版本使用 field_obj.rel.model.objects.all() db_condition = self.get_db_condition(request, *args, **kwargs) return SearchGroupRow(field_obj.related_model.objects.filter(**db_condition), self) # 这里得到的是 Queryset 类型 else: # 获取 choice 的数据 field_obj.choices self.is_choice = True return SearchGroupRow(field_obj.choices, self) # 这里得到的是 tuple 类型
在 UserInfoHandler 这个类中, 为search_group 添加 对象的时候。 可以选择性的为, Option类,添加一个函数进去。 如果添加了那么就会执行,传入的这个函数。 页面进行显示的时候, 就能够根据,用户自己的函数,逻辑进行展示。
class UserInfoHandler(StartHandler):
...................# 配置组合搜索,想要展示哪些, 搜索的条件 search_group = [ Option("gender",text_func=lambda field_obj:field_obj[1]+"666"), Option("classes"), MyOption("depart",text_func=lambda x:x.title + "123"), ]
最后就是,原来的 __iter__ 改成啥样了呢?
class SearchGroupRow(object): def __init__(self, queryset_or_tuple, option): self.option = option self.queryset_or_tuple = queryset_or_tuple def __iter__(self): '''默认显示。 用户可以自定制''' for item in self.queryset_or_tuple: text = self.option.get_text_func(item) yield "<a href='#'>%s</a>" % text
变简单了!不再进行判断类型。 因为这个工作,前面已经做了。 这里就只是执行了一下 传进来 option 对象下的 get_text_func()这个函数。 并且将当前循环的 field 对象。 传给了他。
最后看一下页面效果:
能看出, gender 我进行了自定制。 后面加了 666
derpart 进行了自定制。 后面加了 123
classes 没有自定制, 默认显示的就是。 模型表中 __str__ 返回的值。
最后一点,就是标题了。 在每个前面,添加上一个标题。 这样就知道,我选择的是什么选项了:
改动也不是很大。 就是通过 verbose_name 获取。 field 对象。 中的 verbose_name 设置的文字。
然后 传给SearchGruop 类。 实例化的时候, 对象就拥有这个参数。
然后 __iter__ 函数中,返回一个这个值就可以了。
class SearchGroupRow(object): def __init__(self, queryset_or_list, option, title): self.title = title self.option = option self.queryset_or_list = queryset_or_list def __iter__(self): for item in self.queryset_or_list: yield self.title yield "<a>全部</a>" text = self.option.get_text_func(item) yield "<a href='#'>%s</a>" % text
class Option(object): '''使用组合搜索, 想要一些自己的搜索条件。 可扩展可继承''' def __init__(self, field, db_condition=None, text_func=None): self.field = field self.db_condition = db_condition self.text_func = text_func if not db_condition: db_condition = {} self.db_condition = db_condition self.is_choice = False # 用于判断field对象里面是一个 choice列表 or 外键的queryset
........................
def get_queryset_or_list(self, model_class, request, *args, **kwargs): '''根据字段去获取数据库关联的数据''' # 根据gender或者classes字符串,组自己对应的model类中,找到字段对象,再根据对象,获取关联的数据 field_obj = model_class._meta.get_field(self.field) # 固定用法mate 类,中的get_field() 就可以根据字符串获取对应的对象 title = field_obj.verbose_name # 获取到,文件的 verbose_name # 对field_obj 的类型做判断。 来确定他是一个 choice 还是一个 foreignkey 外键 if isinstance(field_obj, ForeignKey) or isinstance(field_obj, ManyToManyField): # 获取关联表中的, 数据 django1.0 版本使用 field_obj.rel.model.objects.all() db_condition = self.get_db_condition(request, *args, **kwargs) return SearchGroupRow(field_obj.related_model.objects.filter(**db_condition), self, title) # 传进来 # 这里得到的是 Queryset 类型 else: # 获取 choice 的数据 field_obj.choices self.is_choice = True return SearchGroupRow(field_obj.choices, self, title) # 这里得到的是 tuple 类型