class JSONEncoder(Encoder):
"""
Encodes a CWR class instance into a JSON.
For this, first the instance is transformed into a dictionary, then dumped
into a JSON.
A bit of additional work is done for handling the dates, which are
transformed into the ISO format.
"""
def __init__(self):
super(JSONEncoder, self).__init__()
self._dict_encoder = FileDictionaryEncoder()
def encode(self, entity):
"""
Encodes the data, creating a JSON structure from an instance from the
domain model.
:param entity: the instance to encode
:return: a JSON structure created from the received data
"""
encoded = self._dict_encoder.encode(entity)
if sys.version_info[0] == 2:
result = json.dumps(encoded, ensure_ascii=False,
default=_iso_handler, encoding='latin1')
else:
# For Python 3
result = json.dumps(encoded, ensure_ascii=False,
default=_iso_handler)
return result
class JSONEncoder(Encoder):
"""
Encodes a CWR class instance into a JSON.
For this, first the instance is transformed into a dictionary, then dumped
into a JSON.
A bit of additional work is done for handling the dates, which are
transformed into the ISO format.
"""
def __init__(self):
super(JSONEncoder, self).__init__()
self._dict_encoder = FileDictionaryEncoder()
def encode(self, entity):
"""
Encodes the data, creating a JSON structure from an instance from the
domain model.
:param entity: the instance to encode
:return: a JSON structure created from the received data
"""
encoded = self._dict_encoder.encode(entity)
if sys.version_info[0] == 2:
result = json.dumps(encoded,
ensure_ascii=False,
default=_iso_handler,
encoding='latin1')
else:
# For Python 3
result = json.dumps(encoded,
ensure_ascii=False,
default=_iso_handler)
return result
def setUp(self):
self._encoder = FileDictionaryEncoder()
class TestFileDictionaryEncoding(unittest.TestCase):
def setUp(self):
self._encoder = FileDictionaryEncoder()
def test_encoded(self):
tag = self._get_file_tag()
transmission = self._get_transmission()
data = CWRFile(tag, transmission)
encoded = self._encoder.encode(data)
tag = encoded['tag']
self.assertEqual(2015, tag['year'])
self.assertEqual(123, tag['sequence_n'])
self.assertEqual('SND', tag['sender'])
self.assertEqual('RCV', tag['receiver'])
self.assertEqual(2.1, tag['version'])
transmission = encoded['transmission']
self.assertEqual('HDR', transmission['header']['record_type'])
self.assertEqual('TRL', transmission['trailer']['record_type'])
self.assertEqual(2, len(transmission['groups']))
group = transmission['groups'][0]
self.assertEqual('GRH', group['group_header']['record_type'])
self.assertEqual('GRT', group['group_trailer']['record_type'])
self.assertEqual(1, len(group['transactions']))
self.assertEqual(4, len(group['transactions'][0]))
self.assertEqual('ACK', group['transactions'][0][0]['record_type'])
def _get_file_tag(self):
return FileTag(year=2015,
sequence_n=123,
sender='SND',
receiver='RCV',
version=2.1)
def _get_transmission(self):
header = TransmissionHeader(
record_type='HDR',
sender_id='ABC334',
sender_name='SENDER',
sender_type='SO',
creation_date_time=datetime.datetime.strptime(
'20030216', '%Y%m%d').date(),
transmission_date=datetime.datetime.strptime('20030217',
'%Y%m%d').date(),
edi_standard='01.10',
character_set='ASCII')
trailer = TransmissionTrailer(record_type='TRL',
group_count=155,
transaction_count=245,
record_count=568)
groups = [self._get_group(), self._get_group()]
return Transmission(header, trailer, groups)
def _get_group(self):
header = GroupHeader(record_type='GRH',
group_id=3,
transaction_type='AGR',
version_number='02.10',
batch_request_id=15)
trailer = GroupTrailer(record_type='GRT',
group_id=3,
transaction_count=15,
record_count=20)
transactions = [self._get_transaction()]
return Group(header, trailer, transactions)
def _get_transaction(self):
acknowledgement = self._get_ack()
message1 = self._get_message()
message2 = self._get_message()
agreement = self._get_agreement()
return [acknowledgement, message1, message2, agreement]
def _get_ack(self):
return AcknowledgementRecord(
record_type='ACK',
transaction_sequence_n=3,
record_sequence_n=15,
original_group_id=4,
original_transaction_sequence_n=5,
original_transaction_type='AGR',
transaction_status='AS',
creation_date_time=datetime.datetime.strptime(
'20030215', '%Y%m%d').date(),
processing_date=datetime.datetime.strptime('20030216',
'%Y%m%d').date(),
creation_title='TITLE',
submitter_creation_n='A123',
recipient_creation_n='B124')
def _get_message(self):
return MessageRecord(record_type='MSG',
transaction_sequence_n=3,
record_sequence_n=15,
message_level='F',
validation_n='AB3',
message_type='G',
message_text='THE MESSAGE',
original_record_sequence_n=124,
message_record_type='AGR')
def _get_agreement(self):
return AgreementRecord(
record_type='ACK',
transaction_sequence_n=3,
record_sequence_n=15,
submitter_agreement_n='AB12',
agreement_type='OS',
agreement_start_date=datetime.datetime.strptime(
'20030215', '%Y%m%d').date(),
number_of_works=12,
prior_royalty_status='D',
post_term_collection_status='D',
international_standard_code='DFG135',
society_assigned_agreement_n='DF35',
sales_manufacture_clause='M',
agreement_end_date=datetime.datetime.strptime(
'20030216', '%Y%m%d').date(),
date_of_signature=datetime.datetime.strptime('20030217',
'%Y%m%d').date(),
retention_end_date=datetime.datetime.strptime(
'20030218', '%Y%m%d').date(),
prior_royalty_start_date=datetime.datetime.strptime(
'20030219', '%Y%m%d').date(),
post_term_collection_end_date=datetime.datetime.strptime(
'20030220', '%Y%m%d').date(),
shares_change=True,
advance_given=True)
def setUp(self):
self._encoder = FileDictionaryEncoder()
class TestFileDictionaryEncoding(unittest.TestCase):
def setUp(self):
self._encoder = FileDictionaryEncoder()
def test_encoded(self):
tag = self._get_file_tag()
transmission = self._get_transmission()
data = CWRFile(tag, transmission)
encoded = self._encoder.encode(data)
tag = encoded['tag']
self.assertEqual(2015, tag['year'])
self.assertEqual(123, tag['sequence_n'])
self.assertEqual('SND', tag['sender'])
self.assertEqual('RCV', tag['receiver'])
self.assertEqual(2.1, tag['version'])
transmission = encoded['transmission']
self.assertEqual('HDR', transmission['header']['record_type'])
self.assertEqual('TRL', transmission['trailer']['record_type'])
self.assertEqual(2, len(transmission['groups']))
group = transmission['groups'][0]
self.assertEqual('GRH', group['group_header']['record_type'])
self.assertEqual('GRT', group['group_trailer']['record_type'])
self.assertEqual(1, len(group['transactions']))
self.assertEqual(4, len(group['transactions'][0]))
self.assertEqual('ACK', group['transactions'][0][0]['record_type'])
def _get_file_tag(self):
return FileTag(year=2015,
sequence_n=123,
sender='SND',
receiver='RCV',
version=2.1)
def _get_transmission(self):
header = TransmissionHeader(record_type='HDR',
sender_id='ABC334',
sender_name='SENDER',
sender_type='SO',
creation_date_time=datetime.datetime.strptime(
'20030216', '%Y%m%d').date(),
transmission_date=datetime.datetime.strptime(
'20030217', '%Y%m%d').date(),
edi_standard='01.10',
character_set='ASCII')
trailer = TransmissionTrailer(record_type='TRL',
group_count=155,
transaction_count=245,
record_count=568)
groups = [self._get_group(), self._get_group()]
return Transmission(header, trailer, groups)
def _get_group(self):
header = GroupHeader(record_type='GRH',
group_id=3,
transaction_type='AGR',
version_number='02.10',
batch_request_id=15)
trailer = GroupTrailer(record_type='GRT',
group_id=3,
transaction_count=15,
record_count=20)
transactions = [self._get_transaction()]
return Group(header, trailer, transactions)
def _get_transaction(self):
acknowledgement = self._get_ack()
message1 = self._get_message()
message2 = self._get_message()
agreement = self._get_agreement()
return [acknowledgement, message1, message2, agreement]
def _get_ack(self):
return AcknowledgementRecord(record_type='ACK',
transaction_sequence_n=3,
record_sequence_n=15,
original_group_id=4,
original_transaction_sequence_n=5,
original_transaction_type='AGR',
transaction_status='AS',
creation_date_time=datetime.datetime.strptime(
'20030215', '%Y%m%d').date(),
processing_date=datetime.datetime.strptime(
'20030216', '%Y%m%d').date(),
creation_title='TITLE',
submitter_creation_n='A123',
recipient_creation_n='B124')
def _get_message(self):
return MessageRecord(record_type='MSG',
transaction_sequence_n=3,
record_sequence_n=15,
message_level='F',
validation_n='AB3',
message_type='G',
message_text='THE MESSAGE',
original_record_sequence_n=124,
message_record_type='AGR')
def _get_agreement(self):
return AgreementRecord(record_type='ACK',
transaction_sequence_n=3,
record_sequence_n=15,
submitter_agreement_n='AB12',
agreement_type='OS',
agreement_start_date=datetime.datetime.strptime(
'20030215', '%Y%m%d').date(),
number_of_works=12,
prior_royalty_status='D',
post_term_collection_status='D',
international_standard_code='DFG135',
society_assigned_agreement_n='DF35',
sales_manufacture_clause='M',
agreement_end_date=datetime.datetime.strptime(
'20030216', '%Y%m%d').date(),
date_of_signature=datetime.datetime.strptime(
'20030217', '%Y%m%d').date(),
retention_end_date=datetime.datetime.strptime(
'20030218', '%Y%m%d').date(),
prior_royalty_start_date=datetime.datetime.strptime(
'20030219', '%Y%m%d').date(),
post_term_collection_end_date=datetime.datetime.strptime(
'20030220', '%Y%m%d').date(),
shares_change=True,
advance_given=True)
def __init__(self):
super(JSONEncoder, self).__init__()
self._dict_encoder = FileDictionaryEncoder()
def __init__(self):
super(JSONEncoder, self).__init__()
self._dict_encoder = FileDictionaryEncoder()