#!/usr/bin/python

import time

import datetime

import pytz

import random

import sys

import argparse

from faker import Faker

from random import randrange

import json

class switch(object):

return False

parser = argparse.ArgumentParser(__file__, description="Fake Websale Transactions Data Generator")

#parser.add_argument("--output", "-o", dest='output_type', help="Write JSON, Avro or ProtoBuf data to a file or to STDOUT", choices=['AVRO','PROTO','JSON','CONSOLE'] )

parser.add_argument("--num", "-n", dest='num_lines', help="Number of lines to generate (0 for infinite)", type=int, default=1)

args = parser.parse_args()

log_lines = args.num_lines

#output_type = args.output_type

faker = Faker()

timestr = time.strftime("%Y%m%d-%H%M%S")

otime = datetime.datetime.now()

cc_types = ['visa','mastercard','amex','discover','diners']

ualist = [faker.firefox(), faker.chrome(), faker.safari(), faker.internet_explorer(), faker.opera()]

service_list = ['manual', 'amazon','dhl','fedex']

cities = ['Hampden', 'Essex', 'Bristol', 'Barnstable', 'Suffolk', 'Norfolk', 'Plymouth', 'Middlesex']

#outFileName = 'data_file_'+timestr+'.data'

f = None

try:

'''

if output_type == 'LOG':

f = open(outFileName,'w')

elif output_type == 'GZ':

f = gzip.open(outFileName+'.gz','w')

elif output_type == 'CONSOLE':

pass

else:

f = sys.stdout

'''

# Create a bunch of session ids

sessions = []

for x in range(0, 100):

sessions.append(faker.sha1(raw_output=False))

# Create a bunch of product ids

product_ids = []

for x in range(0, 100):

product_ids.append(faker.msisdn())

f = sys.stdout

flag = True

while (flag):

cc_type = random.choice(cc_types)

purchase_amount = round(random.uniform(1, 1000),2)

billing_address = {}

billing_address['address1'] = faker.street_address()

billing_address['address2'] = faker.secondary_address()

billing_address['city'] = random.choice(cities)

billing_address['company'] = faker.company()

creditcard_info = {}

creditcard_info['transaction_date'] = faker.date_this_year(before_today=True, after_today=False).strftime("%m/%d/%Y")

creditcard_info['card_number'] = faker.credit_card_number(card_type=cc_type)

creditcard_info['card_expiry_date'] = faker.credit_card_expire(start="now", end="+10y", date_format="%m/%y")

creditcard_info['card_security_code'] = faker.credit_card_security_code(card_type=cc_type)

creditcard_info['purchase_amount'] = purchase_amount

creditcard_info['description'] = faker.company()

client_details = {}

client_details['browser_ip'] = faker.ipv4()

client_details['user_agent'] = random.choice(ualist)

client_details['session_id'] = random.choice(sessions)

fulfillment = {}

fulfillment['fulfillable_quantity'] = 1

fulfillment['fulfillment_service'] = random.choice(service_list)

fulfillment['fulfillment_status'] = 'fulfilled'

fulfillment['grams'] = round(random.uniform(1, 1000),2)

fulfillment['id'] = faker.ean13()

fulfillment['total_price'] = round(purchase_amount,2)

products = {}

products['product_id'] = random.choice(product_ids)

products['quantity'] = random.randint(1, 10)

products['requires_shipping'] = 1

fulfillment['products'] = products

shopper_profile = {}

shopper_profile['name'] = faker.name_female()

shopper_profile['username'] = faker.user_name()

shopper_profile['sex']='F'

shopper_profile['mail']= faker.ascii_safe_email()

order = {}

order['billing_address'] = billing_address

order['credit_card'] = creditcard_info

order['web_client_details'] = client_details

order['fulfillment'] = fulfillment

order['shopper'] = shopper_profile

order['buyer_accepts_marketing'] = faker.boolean(chance_of_getting_true=50)

order['cart_token'] = faker.uuid4()

print(json.dumps(order))

#f.write(json.dumps(order))

#producer = KafkaProducer(bootstrap_servers='localhost:9092', value_serializer=json.loads)

#pkt = json.dumps(order)

#producer.send('TestRun', pkt)

log_lines = log_lines - 1

flag = False if log_lines == 0 else True

finally:

if f is not None:

f.close()