replace usages of py.log

Author: asottile

changelog/822.trivial.rst

@@ -0,0 +1 @@
+Replace internal usage of ``py.log``.

setup.cfg

@@ -40,7 +40,7 @@ project_urls =
 packages = find:
 package_dir = =src
 zip_safe = False
-python_requires = >=3.6
+python_requires = >=3.6.1
 install_requires =
     execnet>=1.1
     pytest>=6.2.0

src/xdist/_producer.py

@@ -0,0 +1,14 @@
+import sys
+from typing import Any, NamedTuple
+
+
+class Producer(NamedTuple):
+    name: str
+    enabled: bool = True
+
+    def __call__(self, *a: Any, **k: Any) -> None:
+        if self.enabled:
+            print(f"[{self.name}]", *a, **k, file=sys.stderr)
+
+    def __getattr__(self, name: str) -> "Producer":
+        return self._replace(name=name)

src/xdist/dsession.py

@@ -1,6 +1,6 @@
-import py
 import pytest
 
+from xdist._producer import Producer
 from xdist.workermanage import NodeManager
 from xdist.scheduler import (
     EachScheduling,
@@ -34,9 +34,7 @@ class DSession:
 
     def __init__(self, config):
         self.config = config
-        self.log = py.log.Producer("dsession")
-        if not config.option.debug:
-            py.log.setconsumer(self.log._keywords, None)
+        self.log = Producer("dsession", enabled=config.option.debug)
         self.nodemanager = None
         self.sched = None
         self.shuttingdown = False

src/xdist/remote.py

@@ -10,11 +10,11 @@
 import os
 import time
 
-import py
 import pytest
 from execnet.gateway_base import dumps, DumpError
 
 from _pytest.config import _prepareconfig, Config
+from xdist._producer import Producer
 
 try:
     from setproctitle import setproctitle
@@ -37,9 +37,7 @@ def __init__(self, config, channel):
         self.config = config
         self.workerid = config.workerinput.get("workerid", "?")
         self.testrunuid = config.workerinput["testrunuid"]
-        self.log = py.log.Producer("worker-%s" % self.workerid)
-        if not config.option.debug:
-            py.log.setconsumer(self.log._keywords, None)
+        self.log = Producer(f"worker-{self.workerid}", enabled=config.option.debug)
         self.channel = channel
         config.pluginmanager.register(self)
 

src/xdist/scheduler/each.py

@@ -1,5 +1,4 @@
-from py.log import Producer
-
+from xdist._producer import Producer
 from xdist.workermanage import parse_spec_config
 from xdist.report import report_collection_diff
 

src/xdist/scheduler/load.py

@@ -1,8 +1,8 @@
 from itertools import cycle
 
-from py.log import Producer
 from _pytest.runner import CollectReport
 
+from xdist._producer import Producer
 from xdist.workermanage import parse_spec_config
 from xdist.report import report_collection_diff
 

src/xdist/scheduler/loadfile.py

@@ -1,5 +1,5 @@
 from .loadscope import LoadScopeScheduling
-from py.log import Producer
+from xdist._producer import Producer
 
 
 class LoadFileScheduling(LoadScopeScheduling):

src/xdist/scheduler/loadgroup.py

@@ -1,5 +1,5 @@
 from .loadscope import LoadScopeScheduling
-from py.log import Producer
+from xdist._producer import Producer
 
 
 class LoadGroupScheduling(LoadScopeScheduling):

src/xdist/scheduler/loadscope.py

@@ -1,7 +1,7 @@
 from collections import OrderedDict
 
 from _pytest.runner import CollectReport
-from py.log import Producer
+from xdist._producer import Producer
 from xdist.report import report_collection_diff
 from xdist.workermanage import parse_spec_config
 

src/xdist/workermanage.py

@@ -9,6 +9,7 @@
 import execnet
 
 import xdist.remote
+from xdist._producer import Producer
 from xdist.plugin import _sys_path
 
 
@@ -230,9 +231,7 @@ def __init__(self, nodemanager, gateway, config, putevent):
         }
         self._down = False
         self._shutdown_sent = False
-        self.log = py.log.Producer("workerctl-%s" % gateway.id)
-        if not self.config.option.debug:
-            py.log.setconsumer(self.log._keywords, None)
+        self.log = Producer(f"workerctl-{gateway.id}", enabled=config.option.debug)
 
     def __repr__(self):
         return "<{} {}>".format(self.__class__.__name__, self.gateway.id)