mirror of https://github.com/snachodog/mybuddy.git
53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
from django.forms import Media
|
|
from typing import Any, Dict, Optional
|
|
from django.forms import Widget
|
|
|
|
class TagsEditor(Widget):
|
|
class Media:
|
|
js = ("babybuddy/js/tags_editor.js",)
|
|
|
|
input_type = 'hidden'
|
|
template_name = 'core/widget_tag_editor.html'
|
|
|
|
@staticmethod
|
|
def __unpack_tag(tag):
|
|
return {'name': tag.name, 'color': tag.color}
|
|
|
|
def format_value(self, value: Any) -> Optional[str]:
|
|
if value is not None and not isinstance(value, str):
|
|
value = [self.__unpack_tag(tag) for tag in value]
|
|
return value
|
|
|
|
def build_attrs(self, base_attrs, extra_attrs=None):
|
|
attrs = super().build_attrs(base_attrs, extra_attrs)
|
|
class_string = attrs.get('class', '')
|
|
class_string = class_string.replace("form-control", "")
|
|
attrs['class'] = class_string + ' babybuddy-tags-editor'
|
|
return attrs
|
|
|
|
def get_context(self, name: str, value: Any, attrs) -> Dict[str, Any]:
|
|
from . import models
|
|
|
|
most_tags = models.BabyBuddyTag.objects.order_by(
|
|
'-last_used'
|
|
).all()[:256]
|
|
|
|
result = super().get_context(name, value, attrs)
|
|
|
|
tag_names = set(x['name'] for x in (result.get('widget', {}).get('value', None) or []))
|
|
quick_suggestion_tags = [
|
|
t for t in most_tags
|
|
if t.name not in tag_names
|
|
][:5]
|
|
|
|
result['widget']['tag_suggestions'] = {
|
|
'quick': [
|
|
self.__unpack_tag(t) for t in quick_suggestion_tags
|
|
if t.name not in tag_names
|
|
],
|
|
'most': [
|
|
self.__unpack_tag(t) for t in most_tags
|
|
]
|
|
}
|
|
return result
|