changed relative imports to absolute; renamed TelegramBuffer.put to TelegramBuffer.append;
This commit is contained in:
		
							parent
							
								
									11672d0512
								
							
						
					
					
						commit
						21334e5a0a
					
				| @ -3,7 +3,7 @@ import asyncio | ||||
| import logging | ||||
| from functools import partial | ||||
| 
 | ||||
| from .protocol import create_dsmr_reader, create_tcp_dsmr_reader | ||||
| from dsmr_parser.protocol import create_dsmr_reader, create_tcp_dsmr_reader | ||||
| 
 | ||||
| 
 | ||||
| def console(): | ||||
|  | ||||
| @ -3,9 +3,9 @@ import re | ||||
| 
 | ||||
| from PyCRC.CRC16 import CRC16 | ||||
| 
 | ||||
| from .objects import MBusObject, MBusObjectV2_2, CosemObject | ||||
| from .exceptions import ParseError, InvalidChecksumError | ||||
| from .obis_references import GAS_METER_READING | ||||
| from dsmr_parser.objects import MBusObject, MBusObjectV2_2, CosemObject | ||||
| from dsmr_parser.exceptions import ParseError, InvalidChecksumError | ||||
| from dsmr_parser.obis_references import GAS_METER_READING | ||||
| 
 | ||||
| logger = logging.getLogger(__name__) | ||||
| 
 | ||||
|  | ||||
| @ -6,10 +6,11 @@ from functools import partial | ||||
| 
 | ||||
| from serial_asyncio import create_serial_connection | ||||
| 
 | ||||
| from . import telegram_specifications | ||||
| from .exceptions import ParseError | ||||
| from .parsers import TelegramParserV2_2, TelegramParserV4 | ||||
| from .serial import (SERIAL_SETTINGS_V2_2, SERIAL_SETTINGS_V4, TelegramBuffer) | ||||
| from dsmr_parser import telegram_specifications | ||||
| from dsmr_parser.exceptions import ParseError | ||||
| from dsmr_parser.parsers import TelegramParserV2_2, TelegramParserV4 | ||||
| from dsmr_parser.serial import (SERIAL_SETTINGS_V2_2, SERIAL_SETTINGS_V4, | ||||
|                                 TelegramBuffer) | ||||
| 
 | ||||
| 
 | ||||
| def create_dsmr_protocol(dsmr_version, telegram_callback, loop=None): | ||||
| @ -79,7 +80,7 @@ class DSMRProtocol(asyncio.Protocol): | ||||
|         """Add incoming data to buffer.""" | ||||
|         data = data.decode('ascii') | ||||
|         self.log.debug('received data: %s', data) | ||||
|         self.telegram_buffer.put(data) | ||||
|         self.telegram_buffer.append(data) | ||||
| 
 | ||||
|         for telegram in self.telegram_buffer.get_all(): | ||||
|             self.handle_telegram(telegram) | ||||
|  | ||||
| @ -59,7 +59,7 @@ class SerialReader(object): | ||||
|         with serial.Serial(**self.serial_settings) as serial_handle: | ||||
|             while True: | ||||
|                 data = serial_handle.readline() | ||||
|                 self.telegram_buffer.put(data.decode('ascii')) | ||||
|                 self.telegram_buffer.append(data.decode('ascii')) | ||||
| 
 | ||||
|                 for telegram in self.telegram_buffer.get_all(): | ||||
|                     try: | ||||
| @ -92,7 +92,7 @@ class AsyncSerialReader(SerialReader): | ||||
|             # Read line if available or give control back to loop until new | ||||
|             # data has arrived. | ||||
|             data = yield from reader.readline() | ||||
|             self.telegram_buffer.put(data.decode('ascii')) | ||||
|             self.telegram_buffer.append(data.decode('ascii')) | ||||
| 
 | ||||
|             for telegram in self.telegram_buffer.get_all(): | ||||
|                 try: | ||||
| @ -122,7 +122,7 @@ class TelegramBuffer(object): | ||||
|             self._remove(telegram) | ||||
|             yield telegram | ||||
| 
 | ||||
|     def put(self, data): | ||||
|     def append(self, data): | ||||
|         """ | ||||
|         Add telegram data to buffer. | ||||
|         :param str data: chars, lines or full telegram strings of telegram data | ||||
|  | ||||
| @ -1,8 +1,8 @@ | ||||
| from decimal import Decimal | ||||
| 
 | ||||
| from . import obis_references as obis | ||||
| from .parsers import CosemParser, ValueParser, MBusParser | ||||
| from .value_types import timestamp | ||||
| from dsmr_parser import obis_references as obis | ||||
| from dsmr_parser.parsers import CosemParser, ValueParser, MBusParser | ||||
| from dsmr_parser.value_types import timestamp | ||||
| 
 | ||||
| 
 | ||||
| """ | ||||
|  | ||||
| @ -10,7 +10,7 @@ class TelegramBufferTest(unittest.TestCase): | ||||
|         self.telegram_buffer = TelegramBuffer() | ||||
| 
 | ||||
|     def test_v22_telegram(self): | ||||
|         self.telegram_buffer.put(TELEGRAM_V2_2) | ||||
|         self.telegram_buffer.append(TELEGRAM_V2_2) | ||||
| 
 | ||||
|         telegram = next(self.telegram_buffer.get_all()) | ||||
| 
 | ||||
| @ -18,7 +18,7 @@ class TelegramBufferTest(unittest.TestCase): | ||||
|         self.assertEqual(self.telegram_buffer._buffer, '') | ||||
| 
 | ||||
|     def test_v42_telegram(self): | ||||
|         self.telegram_buffer.put(TELEGRAM_V4_2) | ||||
|         self.telegram_buffer.append(TELEGRAM_V4_2) | ||||
| 
 | ||||
|         telegram = next(self.telegram_buffer.get_all()) | ||||
| 
 | ||||
| @ -26,7 +26,7 @@ class TelegramBufferTest(unittest.TestCase): | ||||
|         self.assertEqual(self.telegram_buffer._buffer, '') | ||||
| 
 | ||||
|     def test_multiple_mixed_telegrams(self): | ||||
|         self.telegram_buffer.put( | ||||
|         self.telegram_buffer.append( | ||||
|             ''.join((TELEGRAM_V2_2, TELEGRAM_V4_2, TELEGRAM_V2_2)) | ||||
|         ) | ||||
| 
 | ||||
| @ -47,7 +47,7 @@ class TelegramBufferTest(unittest.TestCase): | ||||
|         # There are unclosed telegrams at the start of the buffer. | ||||
|         incomplete_telegram = TELEGRAM_V4_2[:-1] | ||||
| 
 | ||||
|         self.telegram_buffer.put(incomplete_telegram + TELEGRAM_V4_2) | ||||
|         self.telegram_buffer.append(incomplete_telegram + TELEGRAM_V4_2) | ||||
| 
 | ||||
|         telegram = next(self.telegram_buffer.get_all()) | ||||
| 
 | ||||
| @ -59,7 +59,7 @@ class TelegramBufferTest(unittest.TestCase): | ||||
|         # the buffer was being filled while the telegram was outputted halfway. | ||||
|         incomplete_telegram = TELEGRAM_V4_2[1:] | ||||
| 
 | ||||
|         self.telegram_buffer.put(incomplete_telegram + TELEGRAM_V4_2) | ||||
|         self.telegram_buffer.append(incomplete_telegram + TELEGRAM_V4_2) | ||||
| 
 | ||||
|         telegram = next(self.telegram_buffer.get_all()) | ||||
| 
 | ||||
| @ -69,7 +69,7 @@ class TelegramBufferTest(unittest.TestCase): | ||||
|     def test_v42_telegram_trailed_by_unclosed_telegram(self): | ||||
|         incomplete_telegram = TELEGRAM_V4_2[:-1] | ||||
| 
 | ||||
|         self.telegram_buffer.put(TELEGRAM_V4_2 + incomplete_telegram) | ||||
|         self.telegram_buffer.append(TELEGRAM_V4_2 + incomplete_telegram) | ||||
| 
 | ||||
|         telegram = next(self.telegram_buffer.get_all()) | ||||
| 
 | ||||
| @ -79,7 +79,7 @@ class TelegramBufferTest(unittest.TestCase): | ||||
|     def test_v42_telegram_trailed_by_unopened_telegram(self): | ||||
|         incomplete_telegram = TELEGRAM_V4_2[1:] | ||||
| 
 | ||||
|         self.telegram_buffer.put(TELEGRAM_V4_2 + incomplete_telegram) | ||||
|         self.telegram_buffer.append(TELEGRAM_V4_2 + incomplete_telegram) | ||||
| 
 | ||||
|         telegram = next(self.telegram_buffer.get_all()) | ||||
| 
 | ||||
| @ -88,7 +88,7 @@ class TelegramBufferTest(unittest.TestCase): | ||||
| 
 | ||||
|     def test_v42_telegram_adding_line_by_line(self): | ||||
|         for line in TELEGRAM_V4_2.splitlines(keepends=True): | ||||
|             self.telegram_buffer.put(line) | ||||
|             self.telegram_buffer.append(line) | ||||
| 
 | ||||
|         telegram = next(self.telegram_buffer.get_all()) | ||||
| 
 | ||||
| @ -97,7 +97,7 @@ class TelegramBufferTest(unittest.TestCase): | ||||
| 
 | ||||
|     def test_v42_telegram_adding_char_by_char(self): | ||||
|         for char in TELEGRAM_V4_2: | ||||
|             self.telegram_buffer.put(char) | ||||
|             self.telegram_buffer.append(char) | ||||
| 
 | ||||
|         telegram = next(self.telegram_buffer.get_all()) | ||||
| 
 | ||||
|  | ||||
		Loading…
	
		Reference in New Issue
	
	Block a user