Code layout change.

[wolnelektury.git] / src / social / forms.py

diff --git a/src/social/forms.py b/src/social/forms.py
new file mode 100755 (executable)
index 0000000..5834ffa
--- /dev/null
+++ b/src/social/forms.py
@@ -0,0 +1,55 @@
+# -*- coding: utf-8 -*-
+# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#
+from django import forms
+from django.utils.translation import ugettext_lazy as _
+
+from catalogue.models import Tag
+from catalogue import utils
+from social.utils import get_set, set_sets
+
+
+class UserSetsForm(forms.Form):
+    def __init__(self, book, user, *args, **kwargs):
+        super(UserSetsForm, self).__init__(*args, **kwargs)
+        self.fields['set_ids'] = forms.ChoiceField(
+            choices=[(tag.id, tag.name) for tag in
+                Tag.objects.filter(category='set', user=user).iterator()],
+        )
+
+
+class ObjectSetsForm(forms.Form):
+    tags = forms.CharField(label=_('Tags (comma-separated)'), required=False,
+                           widget=forms.Textarea())
+
+    def __init__(self, obj, user, *args, **kwargs):
+        self._obj = obj
+        self._user = user
+        data = kwargs.setdefault('data', {})
+        if 'tags' not in data and user.is_authenticated():
+            data['tags'] = ', '.join(t.name
+                for t in obj.tags.filter(category='set', user=user).iterator() if t.name)
+        super(ObjectSetsForm, self).__init__(*args, **kwargs)
+
+    def save(self, request):
+        tags = [get_set(self._user, tag_name.strip())
+                    for tag_name in self.cleaned_data['tags'].split(',')]
+        set_sets(self._user, self._obj, tags)
+        return {"like": True}
+
+
+class NewSetForm(forms.Form):
+    name = forms.CharField(max_length=50, required=True)
+
+    def __init__(self, *args, **kwargs):
+        super(NewSetForm, self).__init__(*args, **kwargs)
+        self.fields['name'].widget.attrs['title'] = _('Name of the new shelf')
+
+    def save(self, user, commit=True):
+        name = self.cleaned_data['name']
+        new_set = Tag(name=name, slug=utils.get_random_hash(name), sort_key=name.lower(),
+            category='set', user=user)
+
+        new_set.save()
+        return new_set