Merge pull request #11 from aiven/adding-user-bets

Added bet producer

Author: ftisiot

.gitignore

@@ -1,2 +1,4 @@
 **/certs
-.vscode
+.vscode
+*.pyc
+venv

README.md

@@ -40,7 +40,7 @@ Where
 * `topic-name`: the Apache Kafka topic name to write to (the topic needs to be pre-created or `kafka.auto_create_topics_enable` parameter enabled)
 * `nr-messages`: the number of messages to send
 * `max-waiting-time`: the maximum waiting time in seconds between messages
-* `subject`: select amongst various subjects: **pizza** is the default one, but you can generate also `userbehaviour`, `stock`, `realstock` (using the yahoo finance apis) and `metric`.
+* `subject`: select amongst various subjects: `pizza` is the default one, but you can generate also `userbehaviour`, `bet`, `stock`, `realstock` (using the yahoo finance apis) and `metric`.
 
 If successfully connected to a Apache Kafka cluster, the command will output a number of messages (`nr-messages` parameter) that are been sent to Apache Kafka in the form
 

main.py

@@ -9,6 +9,8 @@
 from stockproducer import StockProvider
 from realstockproducer import RealStockProvider
 from metricproducer import MetricProvider
+from userbets import UserBetsProvider
+
 
 MAX_NUMBER_PIZZAS_IN_ORDER = 10
 MAX_ADDITIONAL_TOPPINGS_IN_PIZZA = 5
@@ -50,11 +52,12 @@ def produce_msgs(cert_folder = '~/kafka-pizza/',
  fake.add_provider(MetricProvider)
  elif subject == 'userbehaviour':
  fake.add_provider(UserBehaviorProvider)
+ elif subject == 'bet':
+ fake.add_provider(UserBetsProvider)
  else:
  fake.add_provider(PizzaProvider)
- 
  while i < nr_messages:
- if subject in ['stock', 'userbehaviour', 'realstock', 'metric']:
+ if subject in ['stock', 'userbehaviour', 'realstock', 'metric','bet']:
  message, key = fake.produce_msg()
  else:
  message, key = fake.produce_msg(fake, i, MAX_NUMBER_PIZZAS_IN_ORDER, MAX_ADDITIONAL_TOPPINGS_IN_PIZZA)

userbets.py

@@ -0,0 +1,64 @@
+import random
+from faker.providers import BaseProvider
+
+MIN_AMOUNT = 2 # Min betting amount
+MAX_AMOUNT = 1000 # Max betting amount
+
+ALPHA = 50 
+
+# Adding a UserBets with 3 methods:
+# * username to retrieve the username,
+# * bet_amount to retrieve the amount
+# * betting_channel_event to retrieve event and channel
+class UserBetsProvider(BaseProvider):
+ 
+ def username(self):
+ valid_usernames = [
+ 'nopineappleonpizza',
+ 'catanzaro99',
+ 'thedoctor',
+ 'bettingexpert01',
+ 'losingmoney66',
+ 'manutd007',
+ 'manutd009',
+ 'citylife1',
+ 'lysa_X',
+ 'aiventest',
+ ]
+ return random.choice(valid_usernames)
+
+ def bet_amount(self):
+ #return int(random.triangular(MIN_AMOUNT, MID_AMOUNT, MAX_AMOUNT))
+ return int((random.paretovariate(ALPHA)-1)*(MAX_AMOUNT - MIN_AMOUNT))+1
+ 
+ def bet_category_event(self):
+ valid_events = [
+ {'category': 'Sport', 'subcategory': 'Football', 'event': 'ManUTD vs Chelsea'},
+ {'category': 'Sport', 'subcategory': 'Box', 'event': 'Chicken Legs vs Power Kick'},
+ {'category': 'Sport', 'subcategory': 'Curling', 'event': 'Italy vs England'},
+ {'category': 'Sport', 'subcategory': 'Netball', 'event': 'Sydney vs Camberra'},
+ {'category': 'Lottery', 'subcategory': 'Bingo', 'event': 'Uk Bingo'},
+ {'category': 'Lottery', 'subcategory': 'WinForLife', 'event': 'Win For Life America'},
+ {'category': 'Event', 'subcategory': 'Music', 'event': 'Rick Astley #1 in World Charts'},
+ {'category': 'Event', 'subcategory': 'Politics', 'event': 'Mickey Mouse new Italian President'},
+ {'category': 'Event', 'subcategory': 'Celebrities', 'event': 'Donald Duck and Marge Simpson Wedding'},
+ ]
+ return random.choice(valid_events)
+ 
+ def produce_msg (self):
+ username = self.username()
+ bet_amount = self.bet_amount()
+ bet_event = self.bet_category_event()
+
+ # message composition
+ message = {
+ 'username': username,
+ 'event': {
+ 'category': bet_event['category'],
+ 'subcategory': bet_event['subcategory'],
+ 'name': bet_event['event']
+ },
+ 'amount': bet_amount
+ }
+ key = {'event': bet_event['event']}
+ return message, key