Implementation

ptulsconv/docparser/adr_entity.py

@@ -44,7 +44,6 @@ adr_field_map = ((['Title', 'PT.Session.Name'], 'Title', str),
                  )
 
 
-
 class ADRLine:
     title: str
     supervisor: str
@@ -107,6 +106,34 @@ class ADRLine:
                    formatter=(lambda x: len(x) > 0))
     ]
 
+    def __init__(self):
+        self.title = None
+        self.supervisor = None
+        self.client = None
+        self.scene = None
+        self.version = None
+        self.reel = None
+        self.start = None
+        self.finish = None
+        self.priority = None
+        self.cue_number = None
+        self.character_id = None
+        self.character_name = None
+        self.actor_name = None
+        self.prompt = None
+        self.reason = None
+        self.requested_by = None
+        self.time_budget_mins = None
+        self.note = None
+        self.spot = None
+        self.shot = None
+        self.effort = False
+        self.tv = False
+        self.tbw = False
+        self.omitted = False
+        self.adlib = False
+        self.optional = False
+
     @classmethod
     def from_event(cls, event: Event) -> Optional['ADRLine']:
         new = cls()

ptulsconv/docparser/tag_mapping.py

@@ -30,31 +30,30 @@ class TagMapping:
     def __init__(self, source: str,
                  target: str,
                  alt: Optional[ContentSource] = None,
-                 formatter=None):
+                 formatter=(lambda x: x)):
         self.source = source
         self.target = target
         self.alternate_source = alt
-        self.formatter = formatter or (lambda x: x)
+        self.formatter = formatter
 
     def apply(self, tags: dict,
               clip_content: str,
               track_content: str,
               session_content: str, to: object) -> bool:
 
-        setter = getattr(to, self.target)
         new_value = None
 
         if self.source in tags.keys():
             new_value = tags[self.source]
-        elif self.alternate_source == 1:
+        elif self.alternate_source == TagMapping.ContentSource.Session:
             new_value = session_content
-        elif self.alternate_source == 2:
+        elif self.alternate_source == TagMapping.ContentSource.Track:
             new_value = track_content
-        elif self.alternate_source == 3:
+        elif self.alternate_source == TagMapping.ContentSource.Clip:
             new_value = clip_content
 
         if new_value is not None:
-            setter(self.formatter(new_value))
+            setattr(to, self.target, self.formatter(new_value))
             return True
         else:
             return False