autopep and some minor changes, created timeline selection ops in sam…

…e fashion as noerw
iluvcapra · Jan 26, 2024 · 00da85c · 00da85c
1 parent 542cd1c
commit 00da85c
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 28 deletions.
diff --git a/ptsl/engine.py b/ptsl/engine.py
@@ -218,13 +218,13 @@ def import_data(self,
         return ImportSessionDataBuilder(self, session_path)
 
     def import_audio(self,
-                        file_list: List[str],
-                        destination_path: str=None,
-                        audio_operations: int=None,
-                        audio_destination: int=None,
-                        audio_location: int=None,
-                        timecode: str=None
-                        ):
+                     file_list: List[str],
+                     destination_path: str = None,
+                     audio_operations: int = None,
+                     audio_destination: int = None,
+                     audio_location: int = None,
+                     timecode: str = None
+                     ):
         """
         Import audio data into the currently-open session.
         location_data needs to be provided regardless if empty.
@@ -235,12 +235,12 @@ def import_audio(self,
                                         location_value=timecode
                                         )
         audio_data = pt.AudioData(file_list=file_list,
-                                    destination_path=destination_path,
-                                    audio_operations=audio_operations,
-                                    audio_destination=audio_destination,
-                                    audio_location=audio_location,
-                                    location_data=spot_data
-                                    )
+                                  destination_path=destination_path,
+                                  audio_operations=audio_operations,
+                                  audio_destination=audio_destination,
+                                  audio_location=audio_location,
+                                  location_data=spot_data
+                                  )
         op = ops.Import(import_type=1, audio_data=audio_data)
         self.client.run(op)
 
@@ -365,7 +365,7 @@ def create_memory_location(self,
         """
         Create a new memory location.
         """
-        if general_properties == None:
+        if general_properties is None:
             general_properties = MemoryLocationProperties(track_visibility=False)
         op = ops.CreateMemoryLocation(
             number=memory_number,
@@ -801,7 +801,7 @@ def set_session_video_rate_pull(self, pull_rate: 'SessionRatePull'):
         """
         op = ops.SetSessionVideoRatePullSettings(video_rate_pull=pull_rate)
         self.client.run(op)
-        
+
     def simple_set_timeline_selection(self, in_time: str):
         """
         Set Selection at Timecode
@@ -810,21 +810,21 @@ def simple_set_timeline_selection(self, in_time: str):
         self.client.run(op)
 
     def create_new_tracks(self,
-                            number_of_tracks: int = None,
-                            track_name: str = None,
-                            track_format: TrackFormat = None,
-                            track_type: 'TrackType' = None,
-                            track_timebase: TrackTimebase = None
-                            ):
+                          number_of_tracks: int = None,
+                          track_name: str = None,
+                          track_format: TrackFormat = None,
+                          track_type: 'TrackType' = None,
+                          track_timebase: TrackTimebase = None
+                          ):
         """
         Create new Tracks
         """
         op = ops.CreateNewTracks(number_of_tracks=number_of_tracks,
-                                    track_name=track_name,
-                                    track_format=track_format,
-                                    track_type=track_type,
-                                    track_timebase=track_timebase
-                                    )
+                                 track_name=track_name,
+                                 track_format=track_format,
+                                 track_type=track_type,
+                                 track_timebase=track_timebase
+                                 )
         self.client.run(op)
 
     def cut(self, special: Optional['AutomationDataOptions'] = None):

diff --git a/ptsl/ops/__init__.py b/ptsl/ops/__init__.py
@@ -84,8 +84,11 @@
     TogglePlayState, ToggleRecordEnable, \
     PlayHalfSpeed, RecordHalfSpeed
 
-from .memory_locations import EditMemoryLocation, GetMemoryLocations, CreateMemoryLocation
 
-from .set_timeline_selection import SetTimelineSelection
+
+from .memory_locations import EditMemoryLocation, \
+    GetMemoryLocations, CreateMemoryLocation
+
+from .timeline_selection import SetTimelineSelection, GetTimelineSelection
 from .create_new_tracks import CreateNewTracks
 
diff --git a/ptsl/ops/set_timeline_selection.py → ptsl/ops/timeline_selection.py b/ptsl/ops/set_timeline_selection.py → ptsl/ops/timeline_selection.py
@@ -1,5 +1,8 @@
 from ptsl.ops import Operation
 
 
+class GetTimelineSelection(Operation):
+    pass
+
 class SetTimelineSelection(Operation):
     pass