Кинотеатры

from docx import Document
from openpyxl import Workbook
from pptx import Presentation
from datetime import datetime, timedelta
 
 
class Seat:
    def __init__(self, row, number):
        self.row = row
        self.number = number
        self.is_occupied = False
 
    def __str__(self):
        return f"{'| ' if self.number == 1 else ''}{self.row}:{self.number} |"
 
 
class Hall:
    def __init__(self, hall_number, rows, seats_per_row):
        self.hall_number = hall_number
        self.seats = [[Seat(row, num) for num in range(1, seats_per_row + 1)] for row in range(1, rows + 1)]
        self.sessions = []
 
    def add_session(self, session):
        self.sessions.append(session)
 
    def get_seat_map(self):
        return [[str(seat) for seat in row] for row in self.seats]
 
    def find_free_seats(self, count):
        free_seats = []
        for row in self.seats:
            consecutive_free = 0
            for seat in row:
                if not seat.is_occupied:
                    consecutive_free += 1
                    if consecutive_free == count:
                        free_seats.append(row)
                        break
                else:
                    consecutive_free = 0
        return free_seats
 
 
class Session:
    def __init__(self, movie_name, start_time, duration):
        self.movie_name = movie_name
        self.start_time = start_time
        self.duration = duration
        self.tickets_sold = []
 
    def sell_ticket(self, seat):
        if not seat.is_occupied:
            seat.is_occupied = True
            self.tickets_sold.append(seat)
            return True
        return False
 
 
class Cinema:
    def __init__(self, name):
        self.name = name
        self.halls = {}
 
    def add_hall(self, hall_number, rows, seats_per_row):
        self.halls[hall_number] = Hall(hall_number, rows, seats_per_row)
 
    def add_session_to_hall(self, hall_number, session):
        if hall_number in self.halls:
            self.halls[hall_number].add_session(session)
 
    def find_next_session(self, movie_name):
        for hall in self.halls.values():
            for session in hall.sessions:
                if session.movie_name == movie_name:
                    return session
        return None
 
    def print_hall_seat_map(self, hall_number):
        if hall_number in self.halls:
            seat_map = self.halls[hall_number].get_seat_map()
            for row in seat_map:
                print(" ".join(row))
 
 
'''
# Пример использования системы
cinema = Cinema("Мой Кинотеатр")
cinema.add_hall(1, 10, 10)  # Зал 1 с 5 рядами и 10 местами в ряду
 
# Создаем сеансы
session1 = Session("Формула 1", "18:00", 120)
cinema.add_session_to_hall(1, session1)
 
# Продаем билеты
seat_to_sell = cinema.halls[1].seats[0][0]  # Место A1
session1.sell_ticket(seat_to_sell)
 
# План зала
print("План зала:")
cinema.print_hall_seat_map(1)
 
# Ближайший сеанс
next_session = cinema.find_next_session("Формула 1")
if next_session:
    print(f"Ближайший сеанс фильма '{next_session.movie_name}' начинается в {next_session.start_time}.")
else:
    print("Сеансов данного фильма нет.")
'''
 
 
class ReportGenerator:
    def __init__(self, cinema):
        self.cinema = cinema
 
    def generate_session_report(self, start_date, end_date, filename):
        """
        Генерирует отчет о сеансах в Word за указанный период.
        """
        document = Document()
        document.add_heading('Отчет о сеансах', 0)
 
        for hall_number, hall in self.cinema.halls.items():
            document.add_heading(f'Зал {hall_number}', level=1)
            for session in hall.sessions:
                if start_date <= session.start_time.date() <= end_date:
                    document.add_paragraph(f"Фильм: {session.movie_name}, Время начала: {session.start_time}")
                    document.add_paragraph(f"Продано билетов: {len(session.tickets_sold)}")
                    document.add_paragraph('---')
 
        document.save(filename)
 
    def generate_occupancy_chart(self, start_time, end_time, filename):
        """
        Генерирует график загруженности кинотеатров в Excel за указанный период времени.
        """
        workbook = Workbook()
        worksheet = workbook.active
 
        # Заголовки столбцов
        worksheet.cell(row=1, column=1, value="Кинотеатр")
        worksheet.cell(row=1, column=2, value="Зал")
        worksheet.cell(row=1, column=3, value="Время")
        worksheet.cell(row=1, column=4, value="Загруженность (%)")
 
        row = 2
        for cinema_name, cinema in self.cinema.items():
            for hall_number, hall in cinema.halls.items():
                for session in hall.sessions:
                    if start_time <= session.start_time <= end_time:
                        worksheet.cell(row=row, column=1, value=cinema_name)
                        worksheet.cell(row=row, column=2, value=hall_number)
                        worksheet.cell(row=row, column=3, value=session.start_time)
                        worksheet.cell(row=row, column=4, value=f"{len(session.tickets_sold) / len(hall.seats) * 100}%")
                        row += 1
 
        workbook.save(filename)
 
    def generate_movie_brochure(self, movie_name, filename):
        """
        Создает рекламный буклет для нового фильма в формате PowerPoint.
        """
        presentation = Presentation()
        blank_slide_layout = presentation.slide_layouts[6]
 
        for cinema_name, cinema in self.cinema.items():
            for hall_number, hall in cinema.halls.items():
                for session in hall.sessions:
                    if session.movie_name == movie_name:
                        slide = presentation.slides.add_slide(blank_slide_layout)
                        slide.shapes.title.text = f"Фильм: {movie_name}"
                        slide.placeholders[1].text = f"Кинотеатр: {cinema_name}, Зал: {hall_number}"
                        slide.placeholders[2].text = f"Время сеанса: {session.start_time}"
 
        presentation.save(filename)
 
 
cinema = Cinema("Мой Кинотеатр")
# Добавляем залы, сеансы и продаем билеты, как в примере задачи 1
 
report_generator = ReportGenerator(cinema)
 
# Генерируем отчет о сеансах за последний месяц
start_date = datetime.today() - timedelta(days=30)
end_date = datetime.today()
report_generator.generate_session_report(start_date, end_date, "сеансы_отчет.docx")
 
# Создаем график загруженности кинотеатров за последние 24 часа
start_time = datetime.now() - timedelta(hours=24)
end_time = datetime.now()
report_generator.generate_occupancy_chart(start_time, end_time, "загруженность_график.xlsx")
 
# Создаем рекламный буклет для нового фильма
report_generator.generate_movie_brochure("Новый фильм", "рекламный_буклет.pptx")