From 00da85c10d12af4f657dac95df6e6abfa080805e Mon Sep 17 00:00:00 2001
From: StephenMachete <stephen.machete@gmail.com>
Date: Sat, 27 Jan 2024 00:02:58 +0100
Subject: [PATCH] autopep and some minor changes, created timeline selection
 ops in same fashion as noerw

---
 ptsl/engine.py                                | 52 +++++++++----------
 ptsl/ops/__init__.py                          |  7 ++-
 ...ine_selection.py => timeline_selection.py} |  3 ++
 3 files changed, 34 insertions(+), 28 deletions(-)
 rename ptsl/ops/{set_timeline_selection.py => timeline_selection.py} (62%)

diff --git a/ptsl/engine.py b/ptsl/engine.py
index a043625..cff648f 100644
--- 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
index 8406930..3b5721f 100644
--- 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 b/ptsl/ops/timeline_selection.py
similarity index 62%
rename from ptsl/ops/set_timeline_selection.py
rename to ptsl/ops/timeline_selection.py
index cabd626..afe103c 100644
--- a/ptsl/ops/set_timeline_selection.py
+++ b/ptsl/ops/timeline_selection.py
@@ -1,5 +1,8 @@
 from ptsl.ops import Operation
 
 
+class GetTimelineSelection(Operation):
+    pass
+
 class SetTimelineSelection(Operation):
     pass
\ No newline at end of file