从外部Python文件导入类时遭遇NameError问题求助

Hey there! Let's break down why you're hitting that NameError and fix it up properly, along with a hidden issue you'd run into next.

What's Causing the Error? #

You're running into a common Python module namespace quirk:

• Modules (like your PM_class.py) have their own separate "scope" of variables. The ThorID you define in your main file lives only in the main file's namespace—PM_class.py can't see it unless you explicitly pass it over.
• Worse, default arguments in class methods (like id=ThorID in your powermeter constructor) get evaluated when the class is defined, not when you create an instance. Since PM_class.py has no idea what ThorID is at that point, it throws a NameError.

Also, heads up: Your split code will hit a second error soon—rm (the resource manager) is a global variable in your single-file code, but PM_class.py doesn't have access to it either. Your connect method will fail with another NameError once you fix the first issue.

Fixes to Try #

Here are two clean, modular solutions that fit well with your plan to add a GUI and more classes later.

Solution 1: Pass Parameters Explicitly (Recommended)

This keeps your powermeter class independent of external global variables, making it easier to reuse or modify later.

First, update PM_class.py:

import pyvisa

class powermeter:
    """Power meter class. Contains parameter values and commands."""
    def __init__(self, wavelength=450, meter_range=200e-3, id=""):
        """Define default parameter values"""
        self.wavelength = wavelength  # nm
        self.meter_range = meter_range  # W
        self.ID = id  # Meter's VISA ID
        self.name = ""  # Meter instance name
        self.rm = pyvisa.ResourceManager()  # Initialize resource manager inside the class

    def connect(self):
        """Open a connection to the power meter"""
        if not self.ID:
            raise ValueError("Power meter ID is missing—can't connect!")
        self.name = self.rm.open_resource(self.ID)

    def set_wavelength(self, wavelength):
        """Set the power meter wavelength (nm)"""
        self.name.write(f"CORR:WAV {wavelength}")  # Using f-strings for cleaner syntax

    def set_range(self, meter_range):
        """Set the power meter range (W)"""
        self.name.write(f"POW:DC:RANG {meter_range}")

    def read(self):
        """Read the power meter output (mW)"""
        return float(self.name.query("READ?")) * 1000

Then update your main file (note the fixed import statement—you don't include .py in imports):

import pyvisa
from PM_class import powermeter  # Fixed import: no .py suffix

rm = pyvisa.ResourceManager()
resource_list = rm.list_resources()

# Thor Labs power meter detection logic
Thor_str = "0x1313::0x8079"
Thor = [i for i, s in enumerate(resource_list) if Thor_str in s]
if len(Thor) == 0:
    print("No Thor Labs power meter detected")
    ThorID = ""
else:
    print("Thor Labs power meter detected")
    ThorID = resource_list[Thor[0]]

# Pass ThorID explicitly when creating the instance
pm = powermeter(id=ThorID)
pm.connect()
pm.set_wavelength(400)

Solution 2: Inject Dependencies (Reuse Existing Resource Manager)

If you want to reuse the rm instance you already created in the main file (to avoid reinitializing the resource manager), pass it to the class too:

Update PM_class.py's constructor:

def __init__(self, rm, wavelength=450, meter_range=200e-3, id=""):
    self.rm = rm  # Use the passed-in resource manager
    self.wavelength = wavelength
    self.meter_range = meter_range
    self.ID = id
    self.name = ""

Then create the instance in your main file like this:

pm = powermeter(rm=rm, id=ThorID)

Why Did the Single-File Version Work? #

In your single-file code, ThorID and rm were global variables in the same namespace as the powermeter class. When the class was defined, it could access those global variables directly. Splitting into modules breaks that shared namespace, which is why you hit errors—but it's also a good thing, because it forces cleaner, more maintainable code!

内容的提问来源于stack exchange，提问作者McKendrigo