Update adr_entity.py

This commit is contained in:
Jamie Hardt
2022-11-04 18:03:29 -07:00
parent 97d6eeda02
commit 8407d31333

View File

@@ -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]
supervisor: Optional[str]
client: Optional[str]
scene: Optional[str]
version: Optional[str]
reel: Optional[str]
start: Optional[Fraction]
finish: Optional[Fraction]
omitted: bool
note: Optional[str]
requested_by: Optional[str]
title: str = ""
supervisor: Optional[str] = None
client: Optional[str] = None
scene: Optional[str] = None
version: Optional[str] = None
reel: Optional[str] = None
start: Fraction = Fraction(0,1)
finish: Fraction = Fraction(0,1)
omitted: bool = False
note: Optional[str] = None
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]
cue_number: Optional[str]
character_id: Optional[str]
character_name: Optional[str]
actor_name: Optional[str]
prompt: Optional[str]
reason: Optional[str]
time_budget_mins: Optional[float]
spot: Optional[str]
shot: Optional[str]
effort: bool
tv: bool
tbw: bool
adlib: bool
optional: bool
priority: Optional[int] = None
cue_number: Optional[str] = None
character_id: Optional[str] = None
character_name: Optional[str] = None
actor_name: Optional[str] = None
prompt: Optional[str] = None
reason: Optional[str] = None
time_budget_mins: Optional[float] = None
spot: Optional[str] = None
shot: Optional[str] = None
effort: bool = False
tv: bool = False
tbw: bool = False
adlib: bool = False
optional: bool = False
tag_mapping = [
@@ -111,30 +109,30 @@ class ADRLine(GenericEvent):
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
# 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