flask8 tested localy some fixes

Author: prathameshnium

Keithley_2400/Backends/IV_K2400_Loop_Backend_v10.py

@@ -42,6 +42,7 @@
 # Changes_done:Working
 # ------------------------------------------------------------------------
 
+import os
 import numpy as np
 import matplotlib.pyplot as plt
 from time import sleep
@@ -60,7 +61,7 @@ def main():
     sleep(2)
 
     i = 0
-    I = []
+    current_values = []
     Volt = []
 
     # user input ----------------------------------
@@ -83,7 +84,7 @@ def IV_Measure(cur):
         sleep(1.5)
         v_meas = keithley_2400.voltage
         sleep(1)
-        I.append(cur * 1e-6)  # Use the actual sourced value
+        current_values.append(cur * 1e-6)  # Use the actual sourced value
         Volt.append(v_meas)
         print(f"{cur * 1e-6:.3e} A  {v_meas:.4f} V")
         i += 1
@@ -92,7 +93,7 @@ def IV_Measure(cur):
     for i1 in np.arange(0, I_range + I_step, I_step):
         IV_Measure(i1)
 
-    df = pd.DataFrame({'I': I, 'V': Volt})
+    df = pd.DataFrame({'I': current_values, 'V': Volt})
     print("\n--- Measurement Complete ---")
     print(df)
 
@@ -104,7 +105,7 @@ def IV_Measure(cur):
     keithley_2400.shutdown()
     print("Keithley 2400 shutdown complete.")
 
-    plt.plot(I, Volt, marker='o', linestyle='-', color='g', label='I-V Data')
+    plt.plot(current_values, Volt, marker='o', linestyle='-', color='g', label='I-V Data')
     plt.xlabel('Current (A)')
     plt.ylabel('Voltage (V)')
     plt.title('I-V Curve')

Keithley_6517B/High_Resistance/Backends/IV_K6517B_L350_T_Control_Backend_v6.py

@@ -194,6 +194,100 @@ def perform_keithley_zero_check(keithley):
 # --- Main Program Execution ---
 
 
+def stabilize_temperature(lakeshore, target_temp):
+    """Actively controls the heater to reach and stabilize at the target temperature."""
+    print(f"\nMoving to start temperature of {target_temp} K using active control...")
+    while True:
+        current_temp = lakeshore.get_temperature(SENSOR_INPUT)
+
+        if current_temp > target_temp + 0.2:  # System is too warm
+            print(f"Cooling... Current: {current_temp:.4f} K > Target: {target_temp} K", end='\r')
+            lakeshore.set_heater_range(HEATER_OUTPUT, 'off')
+        else:  # System is too cold or within tolerance
+            print(f"Heating... Current: {current_temp:.4f} K <= Target: {target_temp} K", end='\r')
+            lakeshore.set_heater_range(HEATER_OUTPUT, 'medium')
+            lakeshore.set_setpoint(HEATER_OUTPUT, target_temp)
+
+        if abs(current_temp - target_temp) < 0.1:  # Stabilization tolerance
+            print(f"\nStabilized at {current_temp:.4f} K. Waiting 5 seconds before starting ramp.")
+            time.sleep(5)
+            break
+        time.sleep(2)
+
+
+def run_measurement_loop(lakeshore, keithley, filename, end_temp, safety_cutoff, source_voltage, delay):
+    """Runs the main data acquisition loop, logging data and updating the plot."""
+    plt.ion()
+    fig, (ax1, ax2) = plt.subplots(2, 1, figsize=(8, 8))
+    fig.suptitle('Live R-T Measurement', fontsize=16)
+    line1, = ax1.plot([], [], 'b-o', markersize=3)
+    ax1.set_xlabel('Elapsed Time (s)')
+    ax1.set_ylabel('Temperature (K)')
+    ax1.set_title('Temperature Ramp Profile')
+    ax1.grid(True, linestyle=':')
+    line2, = ax2.plot([], [], 'r-s', markersize=3)
+    ax2.set_xlabel('Temperature (K)')
+    ax2.set_ylabel('Resistance (Ω)')
+    ax2.set_title('Resistance vs. Temperature')
+    ax2.grid(True, linestyle=':')
+    ax2.set_yscale('log')
+    fig.tight_layout(rect=[0, 0.03, 1, 0.95])
+    time_data, temp_data, res_data = [], [], []
+
+    start_time = time.time()
+
+    with open(filename, 'w', newline='') as file:
+        writer = csv.writer(file)
+        writer.writerow(["Timestamp", "Elapsed Time (s)", "Temperature (K)", "Heater Output (%)",
+                         "Applied Voltage (V)", "Measured Current (A)", "Resistance (Ohm)"])
+
+    while True:
+        elapsed_time = time.time() - start_time
+        current_temp = lakeshore.get_temperature(SENSOR_INPUT)
+        heater_output = lakeshore.get_heater_output(HEATER_OUTPUT)
+
+        time.sleep(delay)
+        resistance = keithley.resistance
+        current = abs(source_voltage / resistance) if resistance != 0 else float('inf')
+
+        status_str = (
+            f"Time: {elapsed_time:7.2f}s | "
+            f"Temp: {current_temp:8.4f}K | "
+            f"Resistance: {resistance:9.3e} Ω"
+        )
+        print(status_str, end='\r')
+
+        with open(filename, 'a', newline='') as file:
+            writer = csv.writer(file)
+            writer.writerow([
+                datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
+                f"{elapsed_time:.2f}", f"{current_temp:.4f}", f"{heater_output:.2f}",
+                f"{source_voltage:.4e}", f"{current:.4e}", f"{resistance:.4e}"
+            ])
+
+        time_data.append(elapsed_time)
+        temp_data.append(current_temp)
+        res_data.append(resistance)
+
+        line1.set_data(time_data, temp_data)
+        ax1.relim()
+        ax1.autoscale_view()
+        line2.set_data(temp_data, res_data)
+        ax2.relim()
+        ax2.autoscale_view()
+        fig.canvas.draw()
+        fig.canvas.flush_events()
+
+        if current_temp >= safety_cutoff:
+            print(f"\n\n!!! SAFETY CUTOFF REACHED at {current_temp:.4f} K (Limit: {safety_cutoff} K) !!!")
+            break
+        if current_temp >= end_temp:
+            print(f"\n\nTarget temperature of {end_temp} K reached.")
+            break
+
+        time.sleep(2)
+
+
 def main():
     """Main function to run the R-T experiment."""
     root = tk.Tk()
@@ -225,128 +319,22 @@ def main():
         keithley = Keithley6517B(KEITHLEY_VISA)
         print(f"Successfully connected to: {keithley.id}")
         perform_keithley_zero_check(keithley)
-
+ 
         keithley.source_voltage = source_voltage
         keithley.current_nplc = 1
         keithley.enable_source()
         print(f"\nKeithley source enabled and set to {source_voltage} V.")
 
         # --- Setup Live Plot ---
         plt.ion()
-        fig, (ax1, ax2) = plt.subplots(2, 1, figsize=(8, 8))
-        fig.suptitle('Live R-T Measurement', fontsize=16)
-        line1, = ax1.plot([], [], 'b-o', markersize=3)
-        ax1.set_xlabel('Elapsed Time (s)')
-        ax1.set_ylabel('Temperature (K)')
-        ax1.set_title('Temperature Ramp Profile')
-        ax1.grid(True, linestyle=':')
-        line2, = ax2.plot([], [], 'r-s', markersize=3)
-        ax2.set_xlabel('Temperature (K)')
-        ax2.set_ylabel('Resistance (Ω)')
-        ax2.set_title('Resistance vs. Temperature')
-        ax2.grid(True, linestyle=':')
-        ax2.set_yscale('log')
-        fig.tight_layout(rect=[0, 0.03, 1, 0.95])
-        time_data, temp_data, res_data = [], [], []
-
-        # --- NEW: Go to Start Temp and Stabilize with Active Control ---
-        print(
-            f"\nMoving to start temperature of {start_temp} K using active control...")
-        while True:
-            current_temp = lakeshore.get_temperature(SENSOR_INPUT)
-
-            # Active heating/cooling logic
-            if current_temp > start_temp + 0.2:  # System is too warm
-                print(
-                    f"Cooling... Current: {current_temp:.4f} K > Target: {start_temp} K",
-                    end='\r')
-                lakeshore.set_heater_range(HEATER_OUTPUT, 'off')
-            else:  # System is too cold or within tolerance
-                print(
-                    f"Heating... Current: {current_temp:.4f} K <= Target: {start_temp} K",
-                    end='\r')
-                lakeshore.set_heater_range(HEATER_OUTPUT, 'medium')
-                lakeshore.set_setpoint(HEATER_OUTPUT, start_temp)
-
-            # Check for stabilization
-            if abs(current_temp - start_temp) < 0.1:  # Stabilization tolerance
-                print(
-                    f"\nStabilized at {current_temp:.4f} K. Waiting 5 seconds before starting ramp.")
-                time.sleep(5)
-                break
-            time.sleep(2)  # Interval for checking stabilization status
+        stabilize_temperature(lakeshore, start_temp)
 
         # --- Start Ramp and Data Logging ---
         lakeshore.setup_ramp(HEATER_OUTPUT, rate)
         lakeshore.set_setpoint(HEATER_OUTPUT, end_temp)
-        # Ensure heater is on for the ramp
         lakeshore.set_heater_range(HEATER_OUTPUT, 'medium')
         print(f"Ramp started towards {end_temp} K at {rate} K/min.")
-
-        start_time = time.time()
-
-        with open(filename, 'w', newline='') as file:
-            writer = csv.writer(file)
-            writer.writerow(["Timestamp",
-                             "Elapsed Time (s)",
-                             "Temperature (K)",
-                             "Heater Output (%)",
-                             "Applied Voltage (V)",
-                             "Measured Current (A)",
-                             "Resistance (Ohm)"])
-
-        # --- Main experiment loop ---
-        while True:
-            elapsed_time = time.time() - start_time
-            current_temp = lakeshore.get_temperature(SENSOR_INPUT)
-            heater_output = lakeshore.get_heater_output(HEATER_OUTPUT)
-
-            time.sleep(delay)
-            resistance = keithley.resistance
-            current = abs(
-                source_voltage /
-                resistance) if resistance != 0 else float('inf')
-
-            status_str = (
-                f"Time: {elapsed_time:7.2f}s | "
-                f"Temp: {current_temp:8.4f}K | "
-                f"Resistance: {resistance:9.3e} Ω"
-            )
-            print(status_str, end='\r')
-
-            with open(filename, 'a', newline='') as file:
-                writer = csv.writer(file)
-                writer.writerow([
-                    datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
-                    f"{elapsed_time:.2f}", f"{current_temp:.4f}", f"{heater_output:.2f}",
-                    f"{source_voltage:.4e}", f"{current:.4e}", f"{resistance:.4e}"
-                ])
-
-            time_data.append(elapsed_time)
-            temp_data.append(current_temp)
-            res_data.append(resistance)
-
-            line1.set_data(time_data, temp_data)
-            ax1.relim()
-            ax1.autoscale_view()
-            line2.set_data(temp_data, res_data)
-            ax2.relim()
-            ax2.autoscale_view()
-            fig.canvas.draw()
-            fig.canvas.flush_events()
-
-            # --- Check for End Conditions ---
-            if current_temp >= safety_cutoff:
-                print(
-                    f"\n\n!!! SAFETY CUTOFF REACHED at {current_temp:.4f} K (Limit: {safety_cutoff} K) !!!")
-                break
-            if current_temp >= end_temp:
-                print(f"\n\nTarget temperature of {end_temp} K reached.")
-                break
-
-            # The main data logging interval. Should be independent of Keithley
-            # delay.
-            time.sleep(2)
+        run_measurement_loop(lakeshore, keithley, filename, end_temp, safety_cutoff, source_voltage, delay)
 
     except ConnectionError as e:
         print(f"\nCould not start experiment due to a connection failure: {e}")

Keithley_6517B/High_Resistance/RT_K6517B_L350_T_Control_GUI_v13.py

@@ -988,6 +988,39 @@ def _process_data_queue(self):
         if self.is_running or self.is_stabilizing:
             self.root.after(200, self._process_data_queue)
 
+    def _handle_new_data_point(self, data):
+        """Helper to process a single data point from the queue."""
+        temp, htr, cur, res, elapsed = data
+        self.log(
+            f"T:{temp:.3f}K | R:{res:.3e}Ω | Htr:{htr:.1f}% ({self.current_heater_range})")
+        with open(self.data_filepath, 'a', newline='') as f:
+            writer = csv.writer(f)
+            writer.writerow(
+                [
+                    datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
+                    f"{elapsed:.2f}",
+                    f"{temp:.4f}",
+                    f"{htr:.2f}",
+                    f"{self.backend.params['source_voltage']:.4e}",
+                    f"{cur:.4e}",
+                    f"{res:.4e}"])
+
+        self.data_storage['time'].append(elapsed)
+        self.data_storage['temperature'].append(temp)
+        self.data_storage['current'].append(cur)
+        self.data_storage['resistance'].append(res)
+
+        # Update plot data
+        self.line_main.set_data(
+            self.data_storage['temperature'],
+            self.data_storage['resistance'])
+        self.line_sub1.set_data(
+            self.data_storage['temperature'],
+            self.data_storage['current'])
+        self.line_sub2.set_data(
+            self.data_storage['time'],
+            self.data_storage['temperature'])
+
     def _scan_for_visa_instruments(self):
         if not pyvisa:
             self.log("ERROR: PyVISA is not installed.")

Keithley_6517B/Pyroelectricity/Backends/Current_K6517B_Simple_Backend_v10.py

@@ -8,8 +8,8 @@
 import pandas as pd
 from pymeasure.instruments.keithley import Keithley6517B
 
-I = []
-t = []
+time_data = []
+current_data = []
 
 try:
     keithley = Keithley6517B("GPIB0::27::INSTR")
@@ -23,17 +23,17 @@
     while True:
         elapsed_time = time.time() - start_time
         current = keithley.current
-        t.append(elapsed_time)
-        I.append(current)
+        time_data.append(elapsed_time)
+        current_data.append(current)
         print(f"Time: {elapsed_time:.2f} | Current: {current} A")
         time.sleep(2)
 
 except KeyboardInterrupt:
     print("\nMeasurement stopped by User.")
 
     # --- THIS IS THE FIX ---
-    if t and I:
-        data_df = pd.DataFrame({"Timestamp": t, "Current (A)": I})
+    if time_data and current_data:
+        data_df = pd.DataFrame({"Timestamp": time_data, "Current (A)": current_data})
         data_df.to_csv("demo_data.dat", index=False)
         print("Data saved to file: demo_data.dat")
 

Utilities/GPIB_Interface_Rescue_Simple_Backened_v2_.py

@@ -12,7 +12,7 @@ def run_rescue(visa_address: str):
     """
     Connects to a hung instrument and attempts to clear its error state.
     """
-    print(f"--- Instrument Rescue Script Initialized ---")
+    print("--- Instrument Rescue Script Initialized ---")
     rm = pyvisa.ResourceManager()
     instrument = None  # Initialize to None to ensure it exists for the 'finally' block