Родословная простая для начинающих

import json
import pprint
fam={'son':['отец Иван', 'мать'], 'father':['отец Кирилл 1', 'мать Пелагея'], 'mother':['отец Пвел', 'мать Дарья']}
#json.dumps(fam)
json_string = json.dumps(fam, indent=4)
with open('fam.json', 'r', encoding='utf-8') as file:
    fam = json.load(file)
#with open('fam.json', 'w', encoding='utf-8') as file:
    #json.dump(fam, file, indent=4)
    pprint.pprint(fam) #строка для кортроля
choice=None
while choice != 0:
    print(
    """
    Выбрать:
    0 - Выйти
    Найти:
    1 - родители сына
    2 - родители отца
    3 - родители матери
    4 - сведения о брате
    41 - найти родителей отца или матери по имени
    Добавить
    5 - сведения о брате
    6 - сведения о сестре
    7 - сведения о другом родственнике
    8 - весь список
    Удалить
    91 - сестру
    92 - родственника
    """
    )
    choice=input('Ваш выбор:')
    print()
    if choice == '0':
        print('Выход')
        break
    elif choice == '1':
        #for i in fam:
            #if i == 'son':
        print('Родители сына: ',fam['son'])
    elif choice == '2':
        for i in fam:
            if i == 'father':
                print('Родители отца: ',*fam[i])
    elif choice == '3':
        for i in fam:
            if i == 'mother':
                print('Родители матери: ',*fam[i])
    elif choice == '4':
        for i in fam:
            if i == 'brother':
                print('Брат ',*fam[i])
    elif choice == '5':
        #brother=input('Кого добавить')
        #if brother not in fam:
        fam['brother']=[input('Вписать ФИО')]
        json_string = json.dumps(fam, indent=4)
        #print(json_string)
        with open("fam.json", "w", encoding="utf-8") as file:
            json.dump(fam, file, indent=4)
    elif choice == '6':
        #brother=input('Кого добавить')
        #if brother not in fam:
        fam['sister']=[input('Вписать ФИО')]
        json_string = json.dumps(fam, indent=4)
        #print(json_string)
        with open("fam.json", "w", encoding="utf-8") as file:
            json.dump(fam, file, indent=4)
    elif choice == '7':
        #genus=input('Кого добавить')
        #if genus not in fam:
        fam['genus']=[input('Вписать ФИО')]
        json_string = json.dumps(fam, indent=4)
        #print(json_string)
        with open("fam.json", "w", encoding="utf-8") as file:
            json.dump(fam, file, indent=4)
    elif choice == '8':
        pprint.pprint(fam)
    elif choice == '91':
        #with open("data.json", "w", encoding="utf-8") as file:
        #    json.dump(fam, file, indent=4)
        del fam['sister']
        json_string = json.dumps(fam, indent=4)
        with open("fam.json", "w", encoding="utf-8") as file:
            json.dump(fam, file, indent=4)
    elif choice == '92':
        del fam['genus']
        json_string = json.dumps(fam, indent=4)
        with open("fam.json", "w", encoding="utf-8") as file:
            json.dump(fam, file, indent=4)
        print(fam)
    elif choice == '41':
        a=input('Ввести имя отца или матери: ')
        if a == 'Иван':
            for k, v in fam.items():
                if 'отец Иван' in v:
                    print('Родители отца: ', fam['father'])
        if a == 'Вера':
            for k, v in fam.items():
                if 'мать Вера' in v:
                    print('Родители матери: ', fam['mother'])
    else:
        print('Вы ошиблись')

# Основной словарь для хранения родословной
# Структура: {имя: {'year': год, 'gender': пол, 'parents': [], 'children': []}}
family_tree = {}
 
def add_person(name, birth_year, gender):
    """Добавление человека в родословную"""
    if name in family_tree:
        print(f"Человек с именем {name} уже существует!")
        return False
    
    family_tree[name] = {
        'birth_year': birth_year,
        'gender': gender,
        'parents': [],
        'children': []
    }
    print(f"Добавлен: {name} ({birth_year} г.р., {gender})")
    return True
 
def set_parents(child_name, parent1, parent2):
    """Установка родителей для ребенка"""
    # Проверка существования всех участников
    for person in [child_name, parent1, parent2]:
        if person not in family_tree:
            print(f"Человек {person} не найден!")
            return False
    
    # Устанавливаем родителей ребенку
    family_tree[child_name]['parents'] = [parent1, parent2]
    
    # Добавляем ребенка родителям
    family_tree[parent1]['children'].append(child_name)
    family_tree[parent2]['children'].append(child_name)
    
    print(f"Родители установлены: {parent1} и {parent2} -> {child_name}")
    return True
 
def find_children(parent_name):
    """Поиск детей человека"""
    if parent_name not in family_tree:
        print(f"Человек {parent_name} не найден!")
        return []
    
    children = family_tree[parent_name]['children']
    if children:
        print(f"Дети {parent_name}: {', '.join(children)}")
    else:
        print(f"У {parent_name} нет детей")
    return children
 
def find_parents(child_name):
    """Поиск родителей человека"""
    if child_name not in family_tree:
        print(f"Человек {child_name} не найден!")
        return []
    
    parents = family_tree[child_name]['parents']
    if parents:
        print(f"Родители {child_name}: {', '.join(parents)}")
    else:
        print(f"Родители {child_name} не указаны")
    return parents
 
def find_siblings(person_name):
    """Поиск братьев и сестер (общие родители)"""
    if person_name not in family_tree:
        print(f"Человек {person_name} не найден!")
        return []
    
    parents = family_tree[person_name]['parents']
    siblings = []
    
    if parents:
        # Собираем всех детей от общих родителей
        for parent in parents:
            parent_children = family_tree[parent]['children']
            siblings.extend([child for child in parent_children 
                           if child != person_name and child not in siblings])
    
    if siblings:
        print(f"Братья/сестры {person_name}: {', '.join(siblings)}")
    else:
        print(f"У {person_name} нет братьев/сестер")
    return siblings
 
def find_grandparents(person_name):
    """Поиск бабушек и дедушек"""
    if person_name not in family_tree:
        print(f"Человек {person_name} не найден!")
        return []
    
    grandparents = []
    parents = family_tree[person_name]['parents']
    
    for parent in parents:
        grandparents.extend(family_tree[parent]['parents'])
    
    if grandparents:
        print(f"Бабушки/дедушки {person_name}: {', '.join(grandparents)}")
    else:
        print(f"У {person_name} нет информации о бабушках/дедушках")
    return grandparents
 
def find_descendants(person_name, level=0):
    """Рекурсивный поиск всех потомков с отступами для визуализации дерева"""
    if person_name not in family_tree:
        return []
    
    descendants = []
    children = family_tree[person_name]['children']
    
    for child in children:
        indent = "  " * level
        print(f"{indent}└── {child}")
        descendants.append(child)
        descendants.extend(find_descendants(child, level + 1))
    
    return descendants
 
def show_family_branch(person_name):
    """Показать ветвь семьи от указанного человека"""
    if person_name not in family_tree:
        print(f"Человек {person_name} не найден!")
        return
    
    print(f"\n=== Ветвь семьи от {person_name} ===")
    print(person_name)
    find_descendants(person_name, 1)
 
def show_person_info(person_name):
    """Показать полную информацию о человеке"""
    if person_name not in family_tree:
        print(f"Человек {person_name} не найден!")
        return
    
    person = family_tree[person_name]
    print(f"\n=== Информация о {person_name} ===")
    print(f"Год рождения: {person['birth_year']}")
    print(f"Пол: {person['gender']}")
    print(f"Родители: {', '.join(person['parents']) if person['parents'] else 'не указаны'}")
    print(f"Дети: {', '.join(person['children']) if person['children'] else 'нет'}")
 
def show_all_people():
    """Показать всех людей в родословной"""
    if not family_tree:
        print("Родословная пуста!")
        return
    
    print("\n=== Все люди в родословной ===")
    for name, info in family_tree.items():
        print(f"{name} ({info['birth_year']} г.р., {info['gender']})")
 
def count_people():
    """Подсчитать количество людей в родословной"""
    count = len(family_tree)
    print(f"Всего людей в родословной: {count}")
    return count
 
def find_people_by_year(year):
    """Найти людей по году рождения"""
    found = [name for name, info in family_tree.items() 
            if info['birth_year'] == year]
    
    if found:
        print(f"Люди родившиеся в {year} году: {', '.join(found)}")
    else:
        print(f"Нет людей родившихся в {year} году")
    return found
 
def find_people_by_gender(gender):
    """Найти людей по полу"""
    found = [name for name, info in family_tree.items() 
            if info['gender'].lower() == gender.lower()]
    
    if found:
        print(f"Люди пола '{gender}': {', '.join(found)}")
    else:
        print(f"Нет людей пола '{gender}'")
    return found
 
def remove_person(person_name):
    """Удалить человека из родословной"""
    if person_name not in family_tree:
        print(f"Человек {person_name} не найден!")
        return False
    
    # Удаляем ссылки у родителей
    for parent in family_tree[person_name]['parents']:
        if person_name in family_tree[parent]['children']:
            family_tree[parent]['children'].remove(person_name)
    
    # Удаляем ссылки у детей
    for child in family_tree[person_name]['children']:
        if person_name in family_tree[child]['parents']:
            family_tree[child]['parents'].remove(person_name)
    
    # Удаляем самого человека
    del family_tree[person_name]
    print(f"Человек {person_name} удален из родословной")
    return True
 
def show_family_tree():
    """Показать все дерево родословной"""
    if not family_tree:
        print("Родословная пуста!")
        return
    
    print("\n=== Полное дерево родословной ===")
    
    # Находим корневых предков (тех, у кого нет родителей)
    roots = [name for name, info in family_tree.items() if not info['parents']]
    
    if not roots:
        print("Нет корневых предков (у всех есть родители)")
        # Берем первого человека как корень для демонстрации
        roots =[list(family_tree.keys())[0]]
    
    for root in roots:
        show_family_branch(root)
 
def main():
    """Главная функция программы"""
    # Добавим тестовые данные для демонстрации
    test_data = [
        ("Иван", "1950", "м"),
        ("Мария", "1952", "ж"),
        ("Петр", "1975", "м"),
        ("Анна", "1978", "ж"),
        ("Сергей", "2000", "м")
    ]
    
    for name, year, gender in test_data:
        add_person(name, year, gender)
    
    # Установим родственные связи
    set_parents("Петр", "Иван", "Мария")
    set_parents("Анна", "Иван", "Мария")
    set_parents("Сергей", "Петр", "Анна")
    
    while True:
        print("\n" + "="*50)
        print("ПРОГРАММА РОДОСЛОВНОЙ (на словарях)")
        print("="*50)
        print("1. Добавить человека")
        print("2. Установить родителей")
        print("3. Найти детей")
        print("4. Найти родителей")
        print("5. Найти братьев/сестер")
        print("6. Найти бабушек/дедушек")
        print("7. Показать информацию о человеке")
        print("8. Показать ветвь семьи")
        print("9. Показать все дерево")
        print("10. Показать всех людей")
        print("11. Найти по году рождения")
        print("12. Найти по полу")
        print("13. Подсчитать людей")
        print("14. Удалить человека")
        print("0. Выход")
        
        choice = input("\nВыберите действие: ").strip()
        
        if choice == '1':
            name = input("Введите имя: ").strip()
            birth_year = input("Введите год рождения: ").strip()
            gender = input("Введите пол (м/ж): ").strip().lower()
            add_person(name, birth_year, gender)
            
        elif choice == '2':
            child = input("Введите имя ребенка: ").strip()
            parent1 = input("Введите имя первого родителя: ").strip()
            parent2 = input("Введите имя второго родителя: ").strip()
            set_parents(child, parent1, parent2)
            
        elif choice == '3':
            name = input("Введите имя человека: ").strip()
            find_children(name)
            
        elif choice == '4':
            name = input("Введите имя человека: ").strip()
            find_parents(name)
            
        elif choice == '5':
            name = input("Введите имя человека: ").strip()
            find_siblings(name)
            
        elif choice == '6':
            name = input("Введите имя человека: ").strip()
            find_grandparents(name)
            
        elif choice == '7':
            name = input("Введите имя человека: ").strip()
            show_person_info(name)
            
        elif choice == '8':
            name = input("Введите имя человека: ").strip()
            show_family_branch(name)
            
        elif choice == '9':
            show_family_tree()
            
        elif choice == '10':
            show_all_people()
            
        elif choice == '11':
            year = input("Введите год рождения: ").strip()
            find_people_by_year(year)
            
        elif choice == '12':
            gender = input("Введите пол для поиска (м/ж): ").strip()
            find_people_by_gender(gender)
            
        elif choice == '13':
            count_people()
            
        elif choice == '14':
            name = input("Введите имя человека для удаления: ").strip()
            remove_person(name)
            
        elif choice == '0':
            print("Выход из программы...")
            print(f"Всего записей в родословной: {len(family_tree)}")
            break
            
        else:
            print("Неверный выбор! Попробуйте снова.")
 
# Запуск программы
if __name__ == "__main__":
    main()

# Основной словарь для хранения родословной
family_tree = {}
 
def add_person(name, birth_year, gender):
    """Добавление человека в родословную"""
    if name in family_tree:
        print("Человек с именем %s уже существует!" % name)
        return False
    
    family_tree[name] = {
        'birth_year': birth_year,
        'gender': gender,
        'parents': [],
        'children': []
    }
    print("Добавлен: %s (%s г.р., %s)" % (name, birth_year, gender))
    return True
 
def set_parents(child_name, parent1, parent2):
    """Установка родителей для ребенка"""
    for person in [child_name, parent1, parent2]:
        if person not in family_tree:
            print("Человек %s не найден!" % person)
            return False
    
    family_tree[child_name]['parents'] = [parent1, parent2]
    family_tree[parent1]['children'].append(child_name)
    family_tree[parent2]['children'].append(child_name)
    
    print("Родители установлены: %s и %s -> %s" % (parent1, parent2, child_name))
    return True
 
def find_children(parent_name):
    """Поиск детей человека"""
    if parent_name not in family_tree:
        print("Человек %s не найден!" % parent_name)
        return []
    
    children = family_tree[parent_name]['children']
    if children:
        print("Дети %s: %s" % (parent_name, ", ".join(children)))
    else:
        print("У %s нет детей" % parent_name)
    return children
 
def find_parents(child_name):
    """Поиск родителей человека"""
    if child_name not in family_tree:
        print("Человек %s не найден!" % child_name)
        return []
    
    parents = family_tree[child_name]['parents']
    if parents:
        print("Родители %s: %s" % (child_name, ", ".join(parents)))
    else:
        print("Родители %s не указаны" % child_name)
    return parents
 
def find_siblings(person_name):
    """Поиск братьев и сестер"""
    if person_name not in family_tree:
        print("Человек %s не найден!" % person_name)
        return []
    
    parents = family_tree[person_name]['parents']
    siblings = []
    
    if parents:
        for parent in parents:
            for child in family_tree[parent]['children']:
                if child != person_name and child not in siblings:
                    siblings.append(child)
    
    if siblings:
        print("Братья/сестры %s: %s" % (person_name, ", ".join(siblings)))
    else:
        print("У %s нет братьев/сестер" % person_name)
    return siblings
 
def find_grandparents(person_name):
    """Поиск бабушек и дедушек"""
    if person_name not in family_tree:
        print("Человек %s не найден!" % person_name)
        return []
    
    grandparents = []
    parents = family_tree[person_name]['parents']
    
    for parent in parents:
        grandparents.extend(family_tree[parent]['parents'])
    
    if grandparents:
        print("Бабушки/дедушки %s: %s" % (person_name, ", ".join(grandparents)))
    else:
        print("У %s нет информации о бабушках/дедушках" % person_name)
    return grandparents
 
def show_family_branch(person_name, level=0):
    """Показать ветвь семьи"""
    if person_name not in family_tree:
        print("Человек %s не найден!" % person_name)
        return
    
    indent = "  " * level
    print("%s%s" % (indent, person_name))
    
    for child in family_tree[person_name]['children']:
        show_family_branch(child, level + 1)
 
def show_person_info(person_name):
    """Показать информацию о человеке"""
    if person_name not in family_tree:
        print("Человек %s не найден!" % person_name)
        return
    
    person = family_tree[person_name]
    print("\n=== Информация о %s ===" % person_name)
    print("Год рождения: %s" % person['birth_year'])
    print("Пол: %s" % person['gender'])
    print("Родители: %s" % (", ".join(person['parents']) if person['parents'] else 'не указаны'))
    print("Дети: %s" % (", ".join(person['children']) if person['children'] else 'нет'))
 
def show_all_people():
    """Показать всех людей"""
    if not family_tree:
        print("Родословная пуста!")
        return
    
    print("\n=== Все люди в родословной ===")
    for name, info in family_tree.items():
        print("%s (%s г.р., %s)" % (name, info['birth_year'], info['gender']))
 
def count_people():
    """Подсчитать количество людей"""
    count = len(family_tree)
    print("Всего людей в родословной: %d" % count)
    return count
 
def find_people_by_year(year):
    """Найти людей по году рождения"""
    found = []
    for name, info in family_tree.items():
        if info['birth_year'] == year:
            found.append(name)
    
    if found:
        print("Люди родившиеся в %s году: %s" % (year, ", ".join(found)))
    else:
        print("Нет людей родившихся в %s году" % year)
    return found
 
def find_people_by_gender(gender):
    """Найти людей по полу"""
    found = []
    for name, info in family_tree.items():
        if info['gender'].lower() == gender.lower():
            found.append(name)
    
    if found:
        print("Люди пола '%s': %s" % (gender, ", ".join(found)))
    else:
        print("Нет людей пола '%s'" % gender)
    return found
 
def remove_person(person_name):
    """Удалить человека"""
    if person_name not in family_tree:
        print("Человек %s не найден!" % person_name)
        return False
    
    # Удаляем ссылки у родителей
    for parent in family_tree[person_name]['parents']:
        if person_name in family_tree[parent]['children']:
            family_tree[parent]['children'].remove(person_name)
    
    # Удаляем ссылки у детей
    for child in family_tree[person_name]['children']:
        if person_name in family_tree[child]['parents']:
            family_tree[child]['parents'].remove(person_name)
    
    del family_tree[person_name]
    print("Человек %s удален" % person_name)
    return True
 
def save_to_file(filename="family_tree.txt"):
    """Сохранить родословную в файл"""
    try:
        with open(filename, 'w', encoding='utf-8') as f:
            for name, info in family_tree.items():
                f.write("%s|%s|%s|%s|%s\n" % (
                    name,
                    info['birth_year'],
                    info['gender'],
                    ",".join(info['parents']),
                    ",".join(info['children'])
                ))
        print("Данные сохранены в файл: %s" % filename)
        return True
    except Exception as e:
        print("Ошибка сохранения: %s" % str(e))
        return False
 
def load_from_file(filename="family_tree.txt"):
    """Загрузить родословную из файла"""
    try:
        with open(filename, 'r', encoding='utf-8') as f:
            for line in f:
                parts = line.strip().split('|')
                if len(parts) == 5:
                    name, year, gender, parents_str, children_str = parts
                    family_tree[name] = {
                        'birth_year': year,
                        'gender': gender,
                        'parents': parents_str.split(",") if parents_str else [],
                        'children': children_str.split(",") if children_str else []
                    }
        print("Данные загружены из файла: %s" % filename)
        return True
    except Exception as e:
        print("Ошибка загрузки: %s" % str(e))
        return False
 
def main():
    """Главная функция"""
    # Пробуем загрузить сохраненные данные
    load_from_file()
    
    while True:
        print("\n" + "="*40)
        print("ПРОГРАММА РОДОСЛОВНОЙ")
        print("="*40)
        print("1. Добавить человека")
        print("2. Установить родителей")
        print("3. Найти детей")
        print("4. Найти родителей")
        print("5. Найти братьев/сестер")
        print("6. Найти бабушек/дедушек")
        print("7. Информация о человеке")
        print("8. Показать ветвь семьи")
        print("9. Показать всех")
        print("10. Найти по году")
        print("11. Найти по полу")
        print("12. Подсчитать")
        print("13. Удалить")
        print("14. Сохранить")
        print("15. Загрузить")
        print("0. Выход")
        
        try:
            choice = raw_input("\nВыберите действие: ").strip()
        except NameError:
            choice = input("\nВыберите действие: ").strip()
        
        if choice == '1':
            name = input("Имя: ").strip()
            birth_year = input("Год рождения: ").strip()
            gender = input("Пол (м/ж): ").strip().lower()
            add_person(name, birth_year, gender)
            
        elif choice == '2':
            child = input("Имя ребенка: ").strip()
            parent1 = input("Первый родитель: ").strip()
            parent2 = input("Второй родитель: ").strip()
            set_parents(child, parent1, parent2)
            
        elif choice == '3':
            name = input("Имя человека: ").strip()
            find_children(name)
            
        elif choice == '4':
            name = input("Имя человека: ").strip()
            find_parents(name)
            
        elif choice == '5':
            name = input("Имя человека: ").strip()
            find_siblings(name)
            
        elif choice == '6':
            name = input("Имя человека: ").strip()
            find_grandparents(name)
            
        elif choice == '7':
            name = input("Имя человека: ").strip()
            show_person_info(name)
            
        elif choice == '8':
            name = input("Имя человека: ").strip()
            print("\nВетвь семьи от %s:" % name)
            show_family_branch(name)
            
        elif choice == '9':
            show_all_people()
            
        elif choice == '10':
            year = input("Год рождения: ").strip()
            find_people_by_year(year)
            
        elif choice == '11':
            gender = input("Пол (м/ж): ").strip()
            find_people_by_gender(gender)
            
        elif choice == '12':
            count_people()
            
        elif choice == '13':
            name = input("Имя для удаления: ").strip()
            remove_person(name)
            
        elif choice == '14':
            save_to_file()
            
        elif choice == '15':
            load_from_file()
            
        elif choice == '0':
            save_to_file()  # Автосохранение при выходе
            print("Выход. Всего записей: %d" % len(family_tree))
            break
            
        else:
            print("Неверный выбор!")
 
if __name__ == "__main__":
    main()