Style work

This commit is contained in:
Jamie Hardt
2019-10-09 11:25:24 -07:00
parent 2cfc0be866
commit 9affade06f

View File

@@ -1,8 +1,10 @@
from parsimonious.nodes import NodeVisitor from parsimonious.nodes import NodeVisitor, Node
class DictionaryParserVisitor(NodeVisitor): class DictionaryParserVisitor(NodeVisitor):
def visit_document(self, node, visited_children):
@staticmethod
def visit_document(node: Node, visited_children) -> dict:
files = next(iter(visited_children[1]), None) files = next(iter(visited_children[1]), None)
clips = next(iter(visited_children[2]), None) clips = next(iter(visited_children[2]), None)
plugins = next(iter(visited_children[3]), None) plugins = next(iter(visited_children[3]), None)
@@ -16,7 +18,8 @@ class DictionaryParserVisitor(NodeVisitor):
tracks=tracks, tracks=tracks,
markers=markers) markers=markers)
def visit_header(self, node, visited_children): @staticmethod
def visit_header(node, visited_children):
tc_drop = False tc_drop = False
for _ in visited_children[20]: for _ in visited_children[20]:
@@ -32,14 +35,17 @@ class DictionaryParserVisitor(NodeVisitor):
count_clips=visited_children[29], count_clips=visited_children[29],
count_files=visited_children[33]) count_files=visited_children[33])
def visit_files_section(self, node, visited_children): @staticmethod
def visit_files_section(node, visited_children):
return list(map(lambda child: dict(filename=child[0], path=child[2]), visited_children[2])) return list(map(lambda child: dict(filename=child[0], path=child[2]), visited_children[2]))
def visit_clips_section(self, node, visited_children): @staticmethod
def visit_clips_section(node, visited_children):
return list(map(lambda child: dict(clip_name=child[0], file=child[2], channel=child[5]), return list(map(lambda child: dict(clip_name=child[0], file=child[2], channel=child[5]),
visited_children[2])) visited_children[2]))
def visit_plugin_listing(self, node, visited_children): @staticmethod
def visit_plugin_listing(node, visited_children):
return list(map(lambda child: dict(manufacturer=child[0], return list(map(lambda child: dict(manufacturer=child[0],
plugin_name=child[2], plugin_name=child[2],
version=child[4], version=child[4],
@@ -48,7 +54,8 @@ class DictionaryParserVisitor(NodeVisitor):
count_instances=child[10]), count_instances=child[10]),
visited_children[2])) visited_children[2]))
def visit_track_block(self, node, visited_children): @staticmethod
def visit_track_block(node, visited_children):
track_header, track_clip_list = visited_children track_header, track_clip_list = visited_children
clips = [] clips = []
for clip in track_clip_list: for clip in track_clip_list:
@@ -69,10 +76,12 @@ class DictionaryParserVisitor(NodeVisitor):
clips=clips clips=clips
) )
def visit_track_listing(selfs, node, visited_children): @staticmethod
def visit_track_listing(node, visited_children):
return visited_children[1] return visited_children[1]
def visit_track_clip_entry(self, node, visited_children): @staticmethod
def visit_track_clip_entry(node, visited_children):
timestamp = None timestamp = None
if isinstance(visited_children[14], list): if isinstance(visited_children[14], list):
timestamp = visited_children[14][0][0] timestamp = visited_children[14][0][0]
@@ -86,16 +95,19 @@ class DictionaryParserVisitor(NodeVisitor):
timestamp=timestamp, timestamp=timestamp,
state=visited_children[15]) state=visited_children[15])
def visit_track_state_list(self, node, visited_children): @staticmethod
def visit_track_state_list(node, visited_children):
states = [] states = []
for next_state in visited_children: for next_state in visited_children:
states.append(next_state[0][0].text) states.append(next_state[0][0].text)
return states return states
def visit_track_clip_state(self, node, visited_children): @staticmethod
def visit_track_clip_state(node, visited_children):
return node.text return node.text
def visit_markers_listing(self, node, visited_children): @staticmethod
def visit_markers_listing(node, visited_children):
markers = [] markers = []
for marker in visited_children[2]: for marker in visited_children[2]:
@@ -103,7 +115,8 @@ class DictionaryParserVisitor(NodeVisitor):
return markers return markers
def visit_marker_record(self, node, visited_children): @staticmethod
def visit_marker_record(node, visited_children):
return dict(number=visited_children[0], return dict(number=visited_children[0],
location=visited_children[3], location=visited_children[3],
time_reference=visited_children[5], time_reference=visited_children[5],
@@ -111,20 +124,23 @@ class DictionaryParserVisitor(NodeVisitor):
name=visited_children[10], name=visited_children[10],
comments=visited_children[12]) comments=visited_children[12])
@staticmethod
def visit_formatted_clip_name(self, node, visited_children): def visit_formatted_clip_name(_, visited_children):
return visited_children[1].text return visited_children[1].text
def visit_string_value(self, node, visited_children): @staticmethod
def visit_string_value(node, visited_children):
return node.text.strip(" ") return node.text.strip(" ")
def visit_integer_value(self, node, visited_children): @staticmethod
def visit_integer_value(node, visited_children):
return int(node.text) return int(node.text)
# def visit_timecode_value(self, node, visited_children): # def visit_timecode_value(self, node, visited_children):
# return node.text.strip(" ") # return node.text.strip(" ")
def visit_float_value(self, node, visited_children): @staticmethod
def visit_float_value(node, visited_children):
return float(node.text) return float(node.text)
def visit_block_ending(self, node, visited_children): def visit_block_ending(self, node, visited_children):