Update adr_entity.py

ptulsconv/docparser/adr_entity.py

@@ -1,5 +1,5 @@
 from ptulsconv.docparser.tag_compiler import Event
-from typing import Optional, List, Tuple, Any
+from typing import Optional, List, Tuple 
 from dataclasses import dataclass
 from fractions import Fraction
 
@@ -11,12 +11,10 @@ def make_entities(from_events: List[Event]) -> Tuple[List['GenericEvent'], List[
     adr_lines = list()
 
     for event in from_events:
-        result: Any = make_entity(event)
+        result = make_entity(event)
         if type(result) is ADRLine:
-            result: ADRLine
             adr_lines.append(result)
         elif type(result) is GenericEvent:
-            result: GenericEvent
             generic_events.append(result)
 
     return generic_events, adr_lines
@@ -41,17 +39,17 @@ def make_entity(from_event: Event) -> Optional[object]:
 
 @dataclass
 class GenericEvent:
-    title: Optional[str]
+    title: str = ""
-    supervisor: Optional[str]
+    supervisor: Optional[str] = None
-    client: Optional[str]
+    client: Optional[str] = None
-    scene: Optional[str]
+    scene: Optional[str] = None
-    version: Optional[str]
+    version: Optional[str] = None
-    reel: Optional[str]
+    reel: Optional[str] = None
-    start: Optional[Fraction]
+    start: Fraction = Fraction(0,1)
-    finish: Optional[Fraction]
+    finish: Fraction = Fraction(0,1)
-    omitted: bool
+    omitted: bool = False
-    note: Optional[str]
+    note: Optional[str] = None
-    requested_by: Optional[str]
+    requested_by: Optional[str] = None
 
     tag_mapping = [
         TagMapping(source='Title', target="title", alt=TagMapping.ContentSource.Session),
@@ -69,21 +67,21 @@ class GenericEvent:
 
 @dataclass
 class ADRLine(GenericEvent):
-    priority: Optional[int]
+    priority: Optional[int] = None
-    cue_number: Optional[str]
+    cue_number: Optional[str] = None
-    character_id: Optional[str]
+    character_id: Optional[str] = None
-    character_name: Optional[str]
+    character_name: Optional[str] = None
-    actor_name: Optional[str]
+    actor_name: Optional[str] = None
-    prompt: Optional[str]
+    prompt: Optional[str] = None
-    reason: Optional[str]
+    reason: Optional[str] = None
-    time_budget_mins: Optional[float]
+    time_budget_mins: Optional[float] = None
-    spot: Optional[str]
+    spot: Optional[str] = None
-    shot: Optional[str]
+    shot: Optional[str] = None
-    effort: bool
+    effort: bool = False
-    tv: bool
+    tv: bool = False
-    tbw: bool
+    tbw: bool = False
-    adlib: bool
+    adlib: bool = False
-    optional: bool
+    optional: bool = False
 
     tag_mapping = [
 
@@ -111,30 +109,30 @@ class ADRLine(GenericEvent):
                    formatter=(lambda x: len(x) > 0))
     ]
 
-    def __init__(self):
+    # def __init__(self):
-        self.title = None
+    #     self.title = None
-        self.supervisor = None
+    #     self.supervisor = None
-        self.client = None
+    #     self.client = None
-        self.scene = None
+    #     self.scene = None
-        self.version = None
+    #     self.version = None
-        self.reel = None
+    #     self.reel = None
-        self.start = None
+    #     self.start = None
-        self.finish = None
+    #     self.finish = None
-        self.priority = None
+    #     self.priority = None
-        self.cue_number = None
+    #     self.cue_number = None
-        self.character_id = None
+    #     self.character_id = None
-        self.character_name = None
+    #     self.character_name = None
-        self.actor_name = None
+    #     self.actor_name = None
-        self.prompt = None
+    #     self.prompt = None
-        self.reason = None
+    #     self.reason = None
-        self.requested_by = None
+    #     self.requested_by = None
-        self.time_budget_mins = None
+    #     self.time_budget_mins = None
-        self.note = None
+    #     self.note = None
-        self.spot = None
+    #     self.spot = None
-        self.shot = None
+    #     self.shot = None
-        self.effort = False
+    #     self.effort = False
-        self.tv = False
+    #     self.tv = False
-        self.tbw = False
+    #     self.tbw = False
-        self.omitted = False
+    #     self.omitted = False
-        self.adlib = False
+    #     self.adlib = False
-        self.optional = False
+    #     self.optional = False