Skip to content

Commit f29262b

Browse files
committed
Improve code readability
1 parent 631826d commit f29262b

2 files changed

Lines changed: 8 additions & 8 deletions

File tree

analytics/client.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -45,15 +45,15 @@ def __init__(self, write_key=None, debug=False, max_queue_size=10000,
4545
self.debug = debug
4646
self.send = send
4747

48-
if transport not in ('http', 's3'):
48+
if transport == 'http':
49+
self.consumer = Consumer(self.queue, write_key, endpoint=endpoint,
50+
on_error=on_error, upload_size=upload_size)
51+
elif transport == 's3':
52+
self.consumer = S3Consumer(self.queue, write_key, endpoint=endpoint,
53+
on_error=on_error, upload_size=upload_size)
54+
else:
4955
raise ValueError("transport should be either http or s3")
5056

51-
self.consumer = Consumer(self.queue, write_key, endpoint=endpoint,
52-
on_error=on_error, upload_size=upload_size) \
53-
if transport=='http' else \
54-
S3Consumer(self.queue, write_key, endpoint=endpoint,
55-
on_error=on_error, upload_size=upload_size)
56-
5757
if debug:
5858
self.log.setLevel(logging.DEBUG)
5959

analytics/s3_consumer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ def next(self):
150150
return written_items
151151

152152
def _upload_request(self, attempt=0):
153-
"""Attempt to upload the data present in the buffer. It will throw AssertionError on failure."""
153+
"""Attempt to upload the data present in the buffer. It will throw Exception on failure."""
154154

155155
bucket = self.s3_details['bucket']
156156
key = self.s3_details['key_template'] % (self.s3_details['part'])

0 commit comments

Comments
 (0)