function scan_from_run_number_or_scan

src/cristallina/utils.py

@@ -24,12 +24,17 @@ def scan_info(run_number,base_path=None,small_data=True):
 
     return scan
 
+def scan_from_run_number_or_scan(input_variable):
+    """Returns scan object for run_number or scan_object itself as input"""
+    if type(input_variable) == SFScanInfo:
+        scan = input_variable
+    else:
+        scan = scan_info(input_variable)
+    return scan
+
 def channel_names(run_number,verbose=False):
     """Prints channel names for a given run_number or scan object"""
-    if type(run_number) == SFScanInfo:
-        scan = run_number
-    else:
-        scan = scan_info(run_number)
+    scan = scan_from_run_number_or_scan(run_number)
     
     channel_list = list(scan[0].keys())