Tod Naturlich/script/family generator.py

#!/usr/bin/env python
'''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)