Add utils.py and update forms.py

Marcel Kawski
1 parent ec9dc7f9
Showing 3 changed files with 134 additions and 4 deletions
collector/storage/forms.py
collector/storage/models.py
collector/storage/utils.py
@@ -3,7 +3,8 @@ import re
 from django.forms import ChoiceField, ModelChoiceField, ModelForm, Textarea, CharField, Form
 from django import forms
  
-from .models import Chunk, Document, Participant, Metadata, Keyword
+from .models import Chunk, Document, Participant, Metadata, Keyword, Magazine, BookWithMultipleAuthors
+from .utils import get_doc_with_type
 from projects.ppc.models import Utterance
 from collector import settings
  
@@ -268,8 +269,36 @@ class MetadataForm(ModelForm):
         }
  
  
-class DocDetailsForm(ModelForm):
+DOC_TYPE_DETAILS_FIELDS = {
+    Document: {
+
+    },
+    Magazine: {
+        'fields': ['title', 'publication_date', 'publication_place', 'number', 'channel', 'original_lang'],
+        'labels': {
+            'title': 'Tytuł',
+            'publication_date': 'Data publikacji',
+            'publication_place': 'Miejsce publikacji',
+            'number': 'Numer',
+            'channel': 'Kanał',
+            'original_lang': 'Język oryginału'
+        }
+    },
+    BookWithMultipleAuthors: {
+        'fields': ['title', 'publication_date', 'publication_place', 'publisher', 'channel', 'original_lang'],
+        'labels': {
+            'title': 'Tytuł',
+            'publication_date': 'Data publikacji',
+            'publication_place': 'Miejsce publikacji',
+            'publisher': 'Wydawca',
+            'channel': 'Kanał',
+            'original_lang': 'Język oryginału'
+        }
+    }
+}
  
+
+class DocDetailsForm(ModelForm):
     def __init__(self, *args, **kwargs):
         super(DocDetailsForm, self).__init__(*args, **kwargs)
         self.fields['title'].widget = Textarea(attrs={'rows': 1})
@@ -292,6 +321,54 @@ class DocDetailsForm(ModelForm):
         }
  
  
+class MagazineDetailsForm(ModelForm):
+    def __init__(self, *args, **kwargs):
+        super(MagazineDetailsForm, self).__init__(*args, **kwargs)
+        self.fields['title'].widget = Textarea(attrs={'rows': 1})
+
+    def save(self, commit=True):
+        magazine = super(MagazineDetailsForm, self).save(commit)
+        magazine.changed = True
+        magazine.save()
+        return magazine
+
+    class Meta:
+        model = Magazine
+        fields = ['title', 'publication_date', 'publication_place', 'number', 'channel', 'original_lang']
+        labels = {
+            'title': 'Tytuł',
+            'publication_date': 'Data publikacji',
+            'publication_place': 'Miejsce publikacji',
+            'number': 'Numer',
+            'channel': 'Kanał',
+            'original_lang': 'Język oryginału'
+        }
+
+
+class BWMADetailsForm(ModelForm):
+    def __init__(self, *args, **kwargs):
+        super(BWMADetailsForm, self).__init__(*args, **kwargs)
+        self.fields['title'].widget = Textarea(attrs={'rows': 1})
+
+    def save(self, commit=True):
+        bwma = super(BWMADetailsForm, self).save(commit)
+        bwma.changed = True
+        bwma.save()
+        return bwma
+
+    class Meta:
+        model = BookWithMultipleAuthors
+        fields = ['title', 'publication_date', 'publication_place', 'publisher', 'channel', 'original_lang']
+        labels = {
+            'title': 'Tytuł',
+            'publication_date': 'Data publikacji',
+            'publication_place': 'Miejsce publikacji',
+            'publisher': 'Wydawca',
+            'channel': 'Kanał',
+            'original_lang': 'Język oryginału'
+        }
+
+
 class SubDocDetailsForm(ModelForm):
  
     def __init__(self, *args, **kwargs):
@@ -387,3 +464,18 @@ class ChunkSplitForm(Form):
         self.fields['text'].widget = Textarea(attrs={'class': 'formset-field',
                                                      'placeholder': 'Tekst akapitu',
                                                      'rows': 5})
+
+
+class DocTypeForm(ModelForm):
+    doc_type = ChoiceField(label='Typ dokumentu')
+
+    def __init__(self, *args, **kwargs):
+        doc_types = kwargs.pop('doc_types')
+        super(DocTypeForm, self).__init__(*args, **kwargs)
+        self.fields['doc_type'].choices = doc_types
+        self.instance = get_doc_with_type(self.instance.id)
+        self.fields['doc_type'].initial = type(self.instance).__name__
+
+    class Meta:
+        model = Document
+        fields = ['doc_type']
@@ -156,7 +156,7 @@ class Document(models.Model):
  
  
 class Magazine(Document):
-    number2 = models.CharField(max_length=50, null=True)
+    number = models.CharField(max_length=50, null=True)
  
     def check_details_filling(self):
         if self.title == '' or self.publication_date is None or self.publication_place == '' or self.number == '' \
@@ -170,7 +170,7 @@ class Magazine(Document):
  
  
 class BookWithMultipleAuthors(Document):
-    publisher2 = models.CharField(max_length=100, null=True)
+    publisher = models.CharField(max_length=100, null=True)
  
     def check_details_filling(self):
         if self.title == '' or self.publication_date is None or self.publication_place == '' or self.publisher == '' \
+from .models import Document, Magazine, BookWithMultipleAuthors, Article, Chapter
+
+DOCUMENT_TYPES_CLASSES = Document, Magazine, BookWithMultipleAuthors
+
+SUBDOCUMENT_TYPES = {  # first subdocument type in tuples are the default ones for this type of the document
+    Document: (Document,),
+    Magazine: (Article,),
+    BookWithMultipleAuthors: (Chapter,)
+}
+
+
+def get_remaining_doc_types_tuple(current_type):
+    doc_types = []
+    for doc_type_model in DOCUMENT_TYPES_CLASSES:
+        if doc_type_model is not current_type:
+            doc_types.append((doc_type_model.__name__, doc_type_model.get_doc_type_display()))
+    return tuple(doc_types)
+
+
+def get_remaining_subdoc_types_tuple(current_parent_type, current_type):
+    doc_types = []
+    for doc_type_model in SUBDOCUMENT_TYPES[current_parent_type]:
+        if doc_type_model is not current_type:
+            doc_types.append((doc_type_model.__name__, doc_type_model.get_doc_type_display()))
+    return tuple(doc_types)
+
+
+def get_doc_with_type(doc_id):
+    all_document_types = list(
+        set(list(DOCUMENT_TYPES_CLASSES) + [dt for dt_list in SUBDOCUMENT_TYPES.values() for dt in dt_list]))
+    all_document_types.append(all_document_types.pop(all_document_types.index(Document)))  # Document at the end
+    for doc_type in all_document_types:
+        try:
+            doc = doc_type.objects.get(id=doc_id)
+            return doc
+        except doc_type.DoesNotExist:
+            continue
+    return None  # TODO