import couchdb
import sys
from .base_handler import BaseHandler
from .custom import faker_options_container
from .helpers import fake2db_logger, lower_str_generator, rnd_id_generator
logger, extra_information = fake2db_logger()
d = extra_information
class Fake2dbCouchdbHandler(BaseHandler):
def fake2db_couchdb_initiator(self, number_of_rows, name=None, custom=None):
'''Main handler for the operation
'''
rows = number_of_rows
db = self.database_caller_creator(name)
if custom:
self.custom_db_creator(rows, db, custom)
sys.exit(0)
self.data_filler_simple_registration(rows, db)
self.data_filler_detailed_registration(rows, db)
self.data_filler_company(rows, db)
self.data_filler_user_agent(rows, db)
self.data_filler_customer(rows, db)
def database_caller_creator(self, name=None):
'''creates a couchdb database
returns the related connection object
which will be later used to spawn the cursor
'''
couch = couchdb.Server()
if name:
db = couch.create(name)
else:
n = 'couchdb_' + lower_str_generator(self)
db = couch.create(n)
logger.warning('couchdb database created with the name: %s', n,
extra=d)
return db
def custom_db_creator(self, number_of_rows, db, custom):
try:
data_list = list()
custom_d = faker_options_container()
for c in custom:
if custom_d.get(c):
logger.warning("fake2db found valid custom key provided: %s" % c, extra=d)
else:
logger.error("fake2db does not support the custom key you provided.", extra=d )
sys.exit(1)
for i in range(0, number_of_rows):
dict_c = {"id": rnd_id_generator(self)}
for c in custom:
dict_c[c] = getattr(self.faker, c)()
db.save(dict_c)
logger.warning('custom Commits are successful after write job!', extra=d)
except Exception as e:
logger.error(e, extra=d)
def data_filler_simple_registration(self, number_of_rows, db):
'''creates and fills the table with simple regis. information
'''
try:
simple_registration = db
for i in range(0, number_of_rows):
post_simple_reg = {
"id": rnd_id_generator(self),
"email": self.faker.safe_email(),
"password": self.faker.md5(raw_output=False)
}
simple_registration.save(post_simple_reg)
logger.warning(
'simple_registration Commits are successful after write job!',
extra=d)
except Exception as e:
logger.error(e, extra=d)
def data_filler_detailed_registration(self, number_of_rows, db):
'''creates and fills the table with detailed regis. information
'''
try:
detailed_registration = db
data_list = list()
for i in range(0, number_of_rows):
post_det_reg = {
"id": rnd_id_generator(self),
"email": self.faker.safe_email(),
"password": self.faker.md5(raw_output=False),
"lastname": self.faker.last_name(),
"name": self.faker.first_name(),
"adress": self.faker.address(),
"phone": self.faker.phone_number()
}
detailed_registration.save(post_det_reg)
logger.warning(
'detailed_registration Commits are successful after write job!',
extra=d)
except Exception as e:
logger.error(e, extra=d)
def data_filler_user_agent(self, number_of_rows, db):
'''creates and fills the table with user agent data
'''
try:
user_agent = db
data_list = list()
for i in range(0, number_of_rows):
post_uo_reg = {
"id": rnd_id_generator(self),
"ip": self.faker.ipv4(),
"countrycode": self.faker.country_code(),
"useragent": self.faker.user_agent()
}
user_agent.save(post_uo_reg)
logger.warning(
'user_agent Commits are successful after write job!',
extra=d)
except Exception as e:
logger.error(e, extra=d)
def data_filler_company(self, number_of_rows, db):
'''creates and fills the table with company data
'''
try:
company = db
data_list = list()
for i in range(0, number_of_rows):
post_comp_reg = {
"id": rnd_id_generator(self),
"name": self.faker.company(),
"date": self.faker.date(pattern="%d-%m-%Y"),
"email": self.faker.company_email(),
"domain": self.faker.safe_email(),
"city": self.faker.city()
}
company.save(post_comp_reg)
logger.warning('companies Commits are successful after write job!',
extra=d)
except Exception as e:
logger.error(e, extra=d)
def data_filler_customer(self, number_of_rows, db):
'''creates and fills the table with customer data
'''
try:
customer = db
data_list = list()
for i in range(0, number_of_rows):
post_cus_reg = {
"id": rnd_id_generator(self),
"name": self.faker.first_name(),
"lastname": self.faker.last_name(),
"address": self.faker.address(),
"country": self.faker.country(),
"city": self.faker.city(),
"registry_date": self.faker.date(pattern="%d-%m-%Y"),
"birthdate": self.faker.date(pattern="%d-%m-%Y"),
"email": self.faker.safe_email(),
"phone_number": self.faker.phone_number(),
"locale": self.faker.locale()
}
customer.save(post_cus_reg)
logger.warning('customer Commits are successful after write job!',
extra=d)
except Exception as e:
logger.error(e, extra=d)
