如何导入文件夹中的所有 python 类？

Question

This is the current structure of my project.

   cities
       newyork.py
       london.py
       berlin.py
   main.py
   requirements.txt

Each of the city files contains a class. For instance Class Newyork: is defined in newyork.py

In my main.py I am importing all the classes for use in this manner:

from cities.newyork import NewYork
from cities.london import London
from cities.berlin import Berlin

What I would like: To have something like this

from cities.* import NewYork, London, Berlin.

Is this possible in python?

Answer 1

Add the following to cities/__init__.py (creating the file if necessary):

from newyork import NewYork
from london import London
from berlin import Berlin

Then you can write

from cities import NewYork, London, Berlin