def _prepare_message_attributes(attributes):
message_attributes = {}
for key, value in attributes.items():
attr_type = MESSAGE_ATTRIBUTES_TYPE[type(value)]
value_key = "BinaryValue" if attr_type == "Binary" else "StringValue"
value = json.dumps(value) if attr_type in ("String.Array",
"Number") else value
message_attributes[key] = {"DataType": attr_type, value_key: value}
return message_attributes
def test_encoder_overwritable():
with overwrite_json_encoder():
# define and register my special encoder for instances of time
@json_encoder.register(datetime.time)
def _(obj):
return repr(obj)
value = datetime.time(1, 2, 3)
result = json.dumps(value)
assert result == '"{}"'.format(repr(value))
def test_encoder_overwritable():
with overwrite_json_encoder():
# define and register my special encoder for instances of time
@json_encoder.register(datetime.time)
def _(obj):
return repr(obj)
value = datetime.time(1, 2, 3)
result = json.dumps(value)
assert result == '"{}"'.format(repr(value))
def checkConfigFile():
home = expanduser("~")
if not os.path.isfile(home + "/config.json"):
with open(home + '/config.json', 'a') as the_file:
init_json = {
'min_time': 3,
"max_time": 10,
"width": 20,
"height": 20
}
init_json_str = json.dumps(init_json)
the_file.write(init_json_str)
async def publish(self,
prefix,
topic,
message_data,
message_attributes=None):
message_attributes = message_attributes or {}
message_attributes = self._prepare_message_attributes(
message_attributes)
if self.dry_run:
logger.info("message=SNSClient.publish called, "
"prefix={}, "
"topic={}, "
"message_data={!r}, "
"message_attributes={!r}".format(
prefix, topic, message_data, message_attributes))
return
topic_arn = await self.get_topic_arn("{}__{}".format(prefix, topic))
message = json.dumps({"default": json.dumps(message_data)})
try:
await self.client.publish(
TopicArn=topic_arn,
MessageStructure="json",
Message=message,
MessageAttributes=message_attributes,
)
except ClientError as exc:
self.handle_exception(exc, prefix, topic, topic_arn)
self.log("published_message, "
"prefix={}, "
"topic={}, "
"arn={}, "
"message={!r}, "
"message_attributes={!r}".format(prefix, topic, topic_arn,
message, message_attributes))
return True
def action_method(self, *args, body=None, params=None, headers=None,
action_name=action_name, **kwargs):
url = self.get_action_full_url(action_name, *args)
method = self.get_action_method(action_name)
if self.json_encode_body and body:
body = json.dumps(body)
request = Request(
url=url,
method=method,
params=params or {},
body=body,
headers=headers or {},
timeout=self.timeout,
kwargs=kwargs
)
request.params.update(self.params)
request.headers.update(self.headers)
return make_request(self.session, request)
async def action_method(self,
*args,
body=None,
params=None,
headers=None,
action_name=action_name):
url = self.get_action_full_url(action_name, *args)
method = self.get_action_method(action_name)
if self.json_encode_body and body:
body = json.dumps(body)
request = Request(url=url,
method=method,
params=params or {},
body=body,
headers=headers or {},
timeout=self.timeout)
request.params.update(self.params)
request.headers.update(self.headers)
async with aiohttp.ClientSession() as session:
return await make_async_request(session, request)
def get_city():
query = "SELECT * from cities"
try:
data = query_result(query, True)
# data = cur.fetchall()
result = []
for item in data:
result.append({
'Sell': item[0],
'List': item[1],
'Living': item[2],
'Rooms': item[3],
'Beds': item[4],
'Baths': item[5],
'Age': item[6],
'Acres': item[7],
'Taxes': item[8]
})
return json.dumps(result)
except Exception as e:
print(e)
return jsonify({'caught': False})
def modify_city():
query = "UPDATE cities \
SET Sell=150 \
WHERE List = 160;"
data = query_result(query)
return json.dumps(data)
def add_city():
query = "INSERT INTO cities \
(Sell, List, Living, Rooms, Beds, Baths, Age, Acres, Taxes) \
VALUES (142, 160, 28, 10, 5, 3, 60, 0.28, 3167);"
data = query_result(query)
return json.dumps(data)
def get_city():
query = "SELECT * from cities"
data = query_result(query)
return json.dumps(data)
def test_encode_datetime():
result = json.dumps(datetime.datetime(2016, 2, 29, 1, 2, 3))
assert result == '"2016-02-29T01:02:03"'
def test_encode_decimal():
result = json.dumps(Decimal('0.001'))
assert result == str('0.001')
def to_representation(self, value):
return json.dumps(value)
def test_encode_uuid():
uuid = 'e3f143b1-445a-4efe-a38b-1b6c9f920932'
result = json.dumps(UUID(uuid))
assert result == '"{}"'.format(uuid)
def test_encode_decimal():
result = json.dumps(Decimal('0.001'))
assert result == str('0.001')
def test_encode_time_with_timezone():
result = json.dumps(
datetime.time(hour=1, minute=2, second=3, microsecond=4, tzinfo=UTC()))
assert result == '"01:02:03.000004+00:00"'
def test_encode_datetime_with_timezone():
result = json.dumps(datetime.datetime(2016, 2, 29, 1, 2, 3, tzinfo=UTC()))
assert result == '"2016-02-29T01:02:03+00:00"'
def to_representation(self, value):
return json.dumps(value)
def insert_data(streamName, data, partitionKey):
kinesis.put_record(StreamName=streamName,
Data=json.dumps(data),
PartitionKey=partitionKey)
def extra_data_dumps(cls, data, validate=True):
form = cls.extra_form_class(data=data)
validate_form(form) if validate else form.full_clean()
return json.dumps(form_data(form))
def delete_city():
query = "DELETE FROM Customers WHERE Sell=150;"
data = query_result(query)
return json.dumps(data)
def test_encode_time():
result = json.dumps(datetime.time(1, 2, 3))
assert result == '"01:02:03"'
def test_encode_uuid():
uuid = 'e3f143b1-445a-4efe-a38b-1b6c9f920932'
result = json.dumps(UUID(uuid))
assert result == '"{}"'.format(uuid)
def test_csv_certificator_meta(csv_certificator, fake_context):
with mock.patch('builtins.open',
mock.mock_open(read_data=json.dumps(fake_context))):
meta = csv_certificator.meta
assert meta == fake_context
def test_encode_date():
result = json.dumps(datetime.date(2016, 2, 29))
assert result == '"2016-02-29"'
def extra_data_dumps(cls, data, validate=True):
form = cls.extra_form_class(data=data)
validate_form(form) if validate else form.full_clean()
return json.dumps(form_data(form))
def test_encode_datetime():
result = json.dumps(datetime.datetime(2016, 2, 29, 1, 2, 3))
assert result == '"2016-02-29T01:02:03"'
def test_encode_time():
result = json.dumps(datetime.time(1, 2, 3))
assert result == '"01:02:03"'
def test_encode_date():
result = json.dumps(datetime.date(2016, 2, 29))
assert result == '"2016-02-29"'
def test_encode_time_with_timezone():
result = json.dumps(datetime.time(
hour=1, minute=2, second=3, microsecond=4, tzinfo=UTC()
))
assert result == '"01:02:03.000004+00:00"'
def test_encode_datetime_with_timezone():
result = json.dumps(datetime.datetime(2016, 2, 29, 1, 2, 3, tzinfo=UTC()))
assert result == '"2016-02-29T01:02:03+00:00"'
