Tod Naturlich/script/family generator.py

Random family generator

Needs the packages "requests" and "termcolor" to work.

import random import requests from termcolor import colored, cprint

GENDERS = ['male', 'female'] BIRTHDAYS = ['January', 'February', 'March', 'April', 'May', 'June',' July', 'August', 'September', 'October', 'November', 'December'] TITLES = ['Parent', 'Sibling']

def generate_age(min_age, max_age, use_gaussian):

   age = 0
   if use_gaussian:
       mean = (max_age + min_age) / 2
       dev = (max_age - min_age) / 4
       age = int(random.gauss(mean, dev))
   else:
       age = random.randint(min_age, max_age)
   return age

def get_school_grade(age):

   if age < 4 or age > 17:
       return None
   if age <= 5:
       return 'Kindergarten'
   if age <= 6:
       return "1st Grade"
   if age <= 7:
       return "2nd Grade"
   if age <= 8:
       return "3rd Grade"
   if age <= 13:
       return "{}th Grade".format(int(age-5))
   if age <= 14:
       return "High School Freshman"
   if age <= 15:
       return "High School Sophomore"
   if age <= 16:
       return "High School Junior"
   return "High School Senior"

def get_names(count=10, gender=None, surname=False, frequency='all'):

   name_params = {
       'count': count,
       'with_surname': 'true' if surname else 'false',
       'frequency': frequency
   }
   if gender and gender in GENDERS:
       name_params['type'] = gender

   response = requests.get('ht' + 'tp://namey.muffinlabs.com/name.json',params=name_params)
   return response.json()

class Character:

   def __init__(self, gender=None, age=None, age_range=None, age_gaussian=False, birthday=None, title=None, alive=True):
       self.alive = alive

       if gender and gender in GENDERS:
           self.gender = gender
       else:
           self.gender = random.choice(GENDERS)

       if age:
           self.age = int(age)
       else:
           if age_range is None:
               age_range = [4, 17]
           self.age = generate_age(age_range[0], age_range[1], age_gaussian)

       if birthday and birthday in BIRTHDAYS:
           self.birthday = birthday
       else:
           self.birthday = random.choice(BIRTHDAYS)

       if title and title in TITLES:
           self.title = title
       else:
           self.title = TITLES[1]

       if title is None or title not in TITLES:
           title = TITLES[1]

       self.grade = get_school_grade(self.age)

   def print_title(self):
       if self.gender == GENDERS[0]:
           if self.title == TITLES[0]:
               return 'Father'
           else:
               return 'Son'
       else:
           if self.title == TITLES[0]:
               return 'Mother'
           else:
               return 'Daughter'

   def print(self, with_names=False):
       names = None
       if with_names:
           print(colored('Getting name list', 'red', attrs=['bold']), end='\r')
           args = {
               'gender': self.gender,
               'surname': True if self.title == TITLES[0] else False,
           }
           names = get_names(**args)
           print(' ' * 80, end='\r')
       description = '{} ({}, {})'.format(self.print_title(), self.age, self.birthday)
       if self.grade:
           description += ' - {}'.format(self.grade)
       if not self.alive:
           description += ' (Deceased)'
       cprint(description, 'blue', attrs=['bold'])
       if with_names:
           cprint(', '.join(names), 'green')

class Family:

   def __init__(self):
       self.no_children = int(random.gauss(3,1))

       self.mom_alive = random.randrange(100) < 90
       self.dad_alive = random.randrange(100) < 95

       self.first_pregnancy_age = int(random.gauss(17, 3))

       self.children = [Character() for i in range(self.no_children)]
       self.children.sort(key=lambda child: child.age, reverse=True)

       self.mom_age = self.first_pregnancy_age + self.children[0].age
       self.dad_age = self.mom_age + int(random.gauss(6, 10))

       self.mom = Character(age=self.mom_age, gender=GENDERS[1], title=TITLES[0], alive=self.mom_alive)
       self.dad = Character(age=self.dad_age, gender=GENDERS[0], title=TITLES[0], alive=self.dad_alive)

   def print(self, with_names=False):
       self.dad.print(with_names)
       self.mom.print(with_names)
       for child in self.children:
           child.print(with_names)

if __name__ == '__main__':

   family = Family()
   family.print(True)