def generate_mockup_data(networks=2, days=7, events_day=10, entries=5, hosts=5,
services=3):
"""Generate mockup data."""
users = models.User.query.all()
for user in users:
sample_services = ['MySQL', 'Chrome', 'Firefox', 'Safari', 'Nginx',
'Apache', 'SSL', 'Android', 'Windows', 'Linux']
for _ in itertools.repeat(None, hosts):
chosen_services = []
for _ in itertools.repeat(None, services):
chosen_services.append(random.choice(sample_services))
models.set_host(user.id,
''.join(random.choice(
string.letters) for _ in range(15)),
chosen_services)
for net in range(networks):
dns2 = '8.8.8.8.6' if net % 2 == 0 else None
models.set_network(str(user.id), 'Home ' + str(net +1),
'eth' + str(net +1),
'FC:AA:25:32:FC:' + str(random.randint(10, 32)),
'1000 Mb/s', 'None',
'192.168.1.' + str(random.randint(1, 254)),
'255.255.255.0', '192.168.1.1', '8.8.8.8.8',
dns2=dns2)
now = datetime.datetime.now()
for day in range(days):
then = now - datetime.timedelta(days=day)
for _ in itertools.repeat(None, events_day):
for network in models.get_user_networks(user.id):
models.save_event(user.id, network['id'],
'Event ' + str(random.randint(1, 100)),
then.strftime(const.STRTIME_DATE),
then.strftime(const.STRTIME_DAY),
then.strftime(const.STRTIME_TIME),
const.PRIORITY_COLOUR[
random.randint(0, 3)])
netids = [net['id'] for net in models.get_user_networks(user.id)]
for let in ['W', 'B']:
models.save_entry(str(user.id), let,
''.join(random.choice(string.letters) for _ in range(15)),
'FC:AA:25:32:FC:' + str(random.randint(10, 32)),
'192.168.1.' + str(random.randint(1, 254)),
[random.choice(netids)])
for _ in itertools.repeat(None, entries - 2):
models.save_entry(str(user.id), random.choice(['B', 'W']),
''.join(random.choice(string.letters) for _ in range(15)),
'FC:AA:25:32:FC:' + str(random.randint(10, 32)),
'192.168.1.' + str(random.randint(1, 254)),
[random.choice(netids)])
