Merge pull request #48 from filips123/fix-updaters-and-empty-timetable

Author: filips123

API/gimvicurnik/database/__init__.py

@@ -13,6 +13,7 @@
     Index,
     Integer,
     SmallInteger,
+    String,
     Text,
     Time,
     func,
@@ -66,7 +67,7 @@ class Document(Base):
     title = Column(Text, nullable=True)
     hash = Column(Text, nullable=True)
     parsed = Column(Boolean, nullable=True)
-    content = Column(Text, nullable=True)
+    content = Column(String(70000), nullable=True)
 
 
 class Entity:
@@ -167,6 +168,9 @@ def get_empty(cls) -> Iterator[Dict[str, Any]]:
         days = (1, 5)
         times = Session.query(func.min(Lesson.time), func.max(Lesson.time))[0]
 
+        if times[0] is None or times[1] is None:
+            return []
+
         classrooms = list(Session.query(Classroom).order_by(Classroom.name))
         lessons = list(Session.query(Lesson).join(Classroom))
 

API/gimvicurnik/updaters/eclassroom.py

@@ -22,6 +22,7 @@
 
 if typing.TYPE_CHECKING:
     from typing import Any, Dict, Iterator, List, Optional
+    from mammoth.documents import Image  # type: ignore
     from sqlalchemy.orm import Session
     from sentry_sdk.tracing import Span
     from ..config import ConfigSourcesEClassroom
@@ -213,7 +214,10 @@ def document_has_content(self, document: DocumentInfo) -> bool:
     def get_content(self, document: DocumentInfo, content: bytes) -> Optional[str]:
         """Get file content of docx circulars."""
 
-        result = convert_to_html(io.BytesIO(content))
+        def ignore_images(_image: Image) -> Dict:
+            return {}
+
+        result = convert_to_html(io.BytesIO(content), convert_image=ignore_images)
         return typing.cast(str, result.value)  # The generated HTML
 
     @with_span(op="parse", pass_span=True)