dgkafka 1.0.0a9__py3-none-any.whl → 1.0.0a10__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
dgkafka/avro_producer.py CHANGED
@@ -84,16 +84,27 @@ class AvroKafkaProducer(KafkaProducer):
84
84
  for k, v in headers.items()]
85
85
 
86
86
  try:
87
- producer.produce(
88
- topic=topic,
89
- value=value,
90
- value_schema=value_schema,
91
- key=key,
92
- key_schema=key_schema,
93
- partition=partition,
94
- on_delivery=self.delivery_report,
95
- headers=headers_list
96
- )
87
+ if not partition:
88
+ producer.produce(
89
+ topic=topic,
90
+ value=value,
91
+ value_schema=value_schema,
92
+ key=key,
93
+ key_schema=key_schema,
94
+ on_delivery=self.delivery_report,
95
+ headers=headers_list
96
+ )
97
+ else:
98
+ producer.produce(
99
+ topic=topic,
100
+ value=value,
101
+ value_schema=value_schema,
102
+ key=key,
103
+ key_schema=key_schema,
104
+ partition=partition,
105
+ on_delivery=self.delivery_report,
106
+ headers=headers_list
107
+ )
97
108
 
98
109
  if flush:
99
110
  producer.flush()
dgkafka/producer.py CHANGED
@@ -119,18 +119,25 @@ class KafkaProducer:
119
119
 
120
120
  # Produce message
121
121
  try:
122
- producer.produce(
123
- topic=topic,
124
- value=value,
125
- key=key_bytes,
126
- partition=partition,
127
- on_delivery=self.delivery_report,
128
- headers=headers_list
129
- )
130
-
122
+ if not partition:
123
+ producer.produce(
124
+ topic=topic,
125
+ value=value,
126
+ key=key_bytes,
127
+ on_delivery=self.delivery_report,
128
+ headers=headers_list
129
+ )
130
+ else:
131
+ producer.produce(
132
+ topic=topic,
133
+ value=value,
134
+ key=key_bytes,
135
+ partition=partition,
136
+ on_delivery=self.delivery_report,
137
+ headers=headers_list
138
+ )
131
139
  if flush:
132
140
  producer.flush()
133
-
134
141
  except Exception as ex:
135
142
  self.logger.error(f"[x] Failed to produce message: {ex}")
136
143
  raise
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: dgkafka
3
- Version: 1.0.0a9
3
+ Version: 1.0.0a10
4
4
  Summary: Kafka clients
5
5
  Home-page: https://gitlab.com/gng-group/dgkafka.git
6
6
  Author: Malanris
@@ -0,0 +1,13 @@
1
+ dgkafka/__init__.py,sha256=fnqVZROyHXipdmhqZaa9XUjvQe795JJKFakwTndAiIw,286
2
+ dgkafka/avro_consumer.py,sha256=By2QWA2BX6ih40x2w3nI1_bv4XtghoVeLcSBNcpJByw,2874
3
+ dgkafka/avro_producer.py,sha256=EJiPjLYNnwXZwcg1mo2RQOw0s1bfiJpwnNFhxywSl-k,4667
4
+ dgkafka/config.py,sha256=uaqllIJZS98_LXg7HE8KFBySZeLJ7BWw2Vv3aft0rtA,3679
5
+ dgkafka/consumer.py,sha256=JoaNnAbCbhSjRl4X4ikReC5Ey5XamJS2DRPyYnP_RRM,4426
6
+ dgkafka/errors.py,sha256=PaH46tXameS--hrrUXKhQkZlBHvMSMPmjhVeRkmFvV0,95
7
+ dgkafka/json_consumer.py,sha256=7Gzn7C9WpyCTPDV6eFDugAx5gC9vdV-HrTh3Nv--zIE,1152
8
+ dgkafka/producer.py,sha256=CupD0qIoJh13dX8A3cZHctWQKqD8m1WRurlTGiIsCoA,5411
9
+ dgkafka-1.0.0a10.dist-info/licenses/LICENSE,sha256=pAZXnNE2dxxwXFIduGyn1gpvPefJtUYOYZOi3yeGG94,1068
10
+ dgkafka-1.0.0a10.dist-info/METADATA,sha256=_KOJOcIMqw4OuVHrZothJA-rt27WgIEEEVuMU980XlQ,6062
11
+ dgkafka-1.0.0a10.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
12
+ dgkafka-1.0.0a10.dist-info/top_level.txt,sha256=GyNrxOh7IPdL0t2SxH8DWxg3fUma-ezQ1Kz4zIr2B7U,8
13
+ dgkafka-1.0.0a10.dist-info/RECORD,,
@@ -1,13 +0,0 @@
1
- dgkafka/__init__.py,sha256=fnqVZROyHXipdmhqZaa9XUjvQe795JJKFakwTndAiIw,286
2
- dgkafka/avro_consumer.py,sha256=By2QWA2BX6ih40x2w3nI1_bv4XtghoVeLcSBNcpJByw,2874
3
- dgkafka/avro_producer.py,sha256=51p9GeavU3M21LfQZuidEkhIpzA-KGQOXhB0xMfNUus,4247
4
- dgkafka/config.py,sha256=uaqllIJZS98_LXg7HE8KFBySZeLJ7BWw2Vv3aft0rtA,3679
5
- dgkafka/consumer.py,sha256=JoaNnAbCbhSjRl4X4ikReC5Ey5XamJS2DRPyYnP_RRM,4426
6
- dgkafka/errors.py,sha256=PaH46tXameS--hrrUXKhQkZlBHvMSMPmjhVeRkmFvV0,95
7
- dgkafka/json_consumer.py,sha256=7Gzn7C9WpyCTPDV6eFDugAx5gC9vdV-HrTh3Nv--zIE,1152
8
- dgkafka/producer.py,sha256=4tKIYUs1zlrie5TOxnL0J38BON2HreOLf2rYW1hVrIs,5085
9
- dgkafka-1.0.0a9.dist-info/licenses/LICENSE,sha256=pAZXnNE2dxxwXFIduGyn1gpvPefJtUYOYZOi3yeGG94,1068
10
- dgkafka-1.0.0a9.dist-info/METADATA,sha256=oBHVvRoOCDkFENPqaV7aK9LYnpgXKT5O_aP7XQFehYI,6061
11
- dgkafka-1.0.0a9.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
12
- dgkafka-1.0.0a9.dist-info/top_level.txt,sha256=GyNrxOh7IPdL0t2SxH8DWxg3fUma-ezQ1Kz4zIr2B7U,8
13
- dgkafka-1.0.0a9.dist-info/RECORD,,