limit what gets imported when importing *

bitelab/config.py

@@ -31,6 +31,8 @@ from pydantic import BaseSettings, Field
 import asyncio
 import aiokq
 
+__all__ = [ 'Settings' ]
+
 # How to deal w/ private vars:
 # https://web.archive.org/web/20201113005838/https://github.com/samuelcolvin/pydantic/issues/655
 class Settings(BaseSettings):

bitelab/mocks.py

@@ -28,6 +28,8 @@
 
 from unittest.mock import AsyncMock, Mock
 
+__all__ = [ 'wrap_subprocess_exec' ]
+
 def wrap_subprocess_exec(mockobj, stdout=b'', stderr=b'', retcode=0):
 	assert isinstance(stdout, bytes)
 	assert isinstance(stderr, bytes)

bitelab/snmp.py

@@ -35,6 +35,8 @@ import asyncio
 import subprocess
 import unittest
 
+__all__ = [ 'snmpget', 'snmpset', 'SNMPPower' ]
+
 async def snmpget(host, oid, type):
 	p = await asyncio.create_subprocess_exec('snmpget', '-Oqv', host, oid,
 	    stdout=subprocess.PIPE)