|
9 | 9 | import logging
|
10 | 10 | import sys
|
11 | 11 | import uuid
|
| 12 | +from logging.handlers import RotatingFileHandler |
12 | 13 |
|
13 | 14 | # Ignore async tests for Python < 3.5
|
14 | 15 | collect_ignore = []
|
|
17 | 18 | collect_ignore.append("features")
|
18 | 19 | collect_ignore.append("examples/async_examples")
|
19 | 20 | else:
|
20 |
| - from tests.asynctests import MockEventProcessor |
| 21 | + sys.path.append("tests") |
| 22 | + from asynctests import MockEventProcessor |
21 | 23 | from azure.eventprocessorhost import EventProcessorHost
|
22 | 24 | from azure.eventprocessorhost import EventHubPartitionPump
|
23 | 25 | from azure.eventprocessorhost import AzureStorageCheckpointLeaseManager
|
|
32 | 34 | from azure.eventhub import EventHubClient, Receiver, Offset
|
33 | 35 |
|
34 | 36 |
|
| 37 | +def get_logger(filename, level=logging.INFO): |
| 38 | + azure_logger = logging.getLogger("azure.eventhub") |
| 39 | + azure_logger.setLevel(level) |
| 40 | + uamqp_logger = logging.getLogger("uamqp") |
| 41 | + uamqp_logger.setLevel(logging.INFO) |
| 42 | + |
| 43 | + formatter = logging.Formatter('%(asctime)s %(name)-12s %(levelname)-8s %(message)s') |
| 44 | + console_handler = logging.StreamHandler(stream=sys.stdout) |
| 45 | + console_handler.setFormatter(formatter) |
| 46 | + if not azure_logger.handlers: |
| 47 | + azure_logger.addHandler(console_handler) |
| 48 | + if not uamqp_logger.handlers: |
| 49 | + uamqp_logger.addHandler(console_handler) |
| 50 | + |
| 51 | + if filename: |
| 52 | + file_handler = RotatingFileHandler(filename, maxBytes=5*1024*1024, backupCount=2) |
| 53 | + file_handler.setFormatter(formatter) |
| 54 | + azure_logger.addHandler(file_handler) |
| 55 | + |
| 56 | + return azure_logger |
| 57 | + |
| 58 | + |
35 | 59 | log = get_logger(None, logging.DEBUG)
|
36 | 60 |
|
37 | 61 |
|
|
0 commit comments