Какая-то хуйня пока, но запушу

alkomkh · Apr 20, 2024 · 44a8083 · 44a8083
1 parent 0eec52d
commit 44a8083
Show file tree

Hide file tree

Showing 3 changed files with 178 additions and 87 deletions.
diff --git a/database.db b/database.db
diff --git a/main.py b/main.py
@@ -38,7 +38,7 @@ def view_pop(view):
         page.go(top_view.route)
 
     page.on_route_change = route_change
-    page.on_view_pop = view_pop
+    # page.on_view_pop = view_pop
     await page.go_async(page.route)
 
 

diff --git a/main_page.py b/main_page.py
@@ -1,5 +1,6 @@
 import flet as ft
 import sqlite3
+import datetime
 controls_list = []
 
 
@@ -9,113 +10,203 @@ async def view(db: sqlite3.Connection, page_: ft.Page):
     async def logout(e):
         page_.session.clear()
         await page_.go_async("/")
+
+    data_columns = []
+    date_list = []
+    date_now = datetime.datetime.now()
 
-    async def add_guest(e):
-        async def close_dlg(e):
-            dlg.open = False
-            await page_.update_async()
+    date = date_now
+
+    for i in range(10):
+        data_columns.append(ft.DataColumn(ft.Text(date)))
+        date_list.append(date)
+        date = date_now + datetime.timedelta(days=i)
 
-        async def add_guest_db(e):
-            cursor.execute(f'''INSERT INTO guests(first_name, last_name) VALUES(?, ?)''', (name_field.value, familia_field.value))
+    rooms = cursor.execute(f"""SELECT * FROM rooms""").fetchall()
+    # print(rooms)
+    rows = []
+    # print(date_list)
+    # print(date_list[0])
+    for room in rooms:
+        cells_list = []
+        for i in date_list:
+            date_str = i.strftime('%Y-%m-%d %H:%M:%S')
+            # res = cursor.execute(f"""SELECT * FROM reservations WHERE room_id = ?""", (room[0],)).fetchall()
+            res = cursor.execute(f"""SELECT * FROM reservations WHERE room_id = ? AND start_date <= ? AND finish_date >= ?""", (room[0], date_str, date_str)).fetchall()
+            # print(res)
+            if res:
+                cells_list.append(ft.DataCell(content=ft.Text("Занято", bgcolor=ft.colors.RED)))
+                print("ZANYATO")
+            else:
+                print("SVOBODNO")
+                cells_list.append(ft.DataCell(content=ft.Text("Свободно", bgcolor=ft.colors.GREEN)))
+
+        rows.append(ft.DataRow(cells=cells_list))
+
+
+
+
+
+
+    table = ft.DataTable(
+        columns=data_columns,
+        rows=rows
+    )
+
+    async def add_res(e):
+
+        async def add(e):
+            cursor.execute(f"""INSERT INTO reservations (room_id, start_date, finish_date, name, surname, passport) VALUES (?, ?, ?, ?, ?, ?)""", (room_id_field.value, date_first.value, date_last.value, name_field.value, surname_field.value, passport_field.value))
             db.commit()
+            dlg.open = False
+            await page_.go_async("/")
+
+
+        async def close(e):
             dlg.open = False
             await page_.update_async()
 
         name_field = ft.TextField(label="Имя")
-        familia_field = ft.TextField(label="Фамилия")
-
-        dlg = ft.AlertDialog(modal=True, 
-                             title=ft.Text("Добавить Постояльца"),
-                             content=ft.Column(controls=[name_field,
-                                                         familia_field,
-                             ], alignment=ft.alignment.center, horizontal_alignment=ft.CrossAxisAlignment.CENTER,
-                                 tight=True),
-                             actions=[ft.TextButton(text="Далее", on_click=add_guest_db), ft.TextButton(text="Отмена", on_click=close_dlg)])
+        surname_field = ft.TextField(label="Фамилия")
+        passport_field = ft.TextField(label="Паспортные данные")
+        room_id_field = ft.TextField(label="Номер номера")
 
-        dlg.open = True
-        page_.dialog = dlg
-        await page_.update_async()
 
-    async def add_reservation(e):
-        global controls_list
-        print("add_reservation")
-        guest_id = None
 
-        async def close_dlg(e):
-            dlg.open = False
-            await page_.update_async()
-
-        async def close_anchor(e):
-            text = f"Color {e.control.data}"
-            print(f"closing view from {text}")
-            guests_selection_dates.close_view(text)
+        date_first = ft.DatePicker(first_date=  datetime.datetime.now(), 
+                                   last_date=datetime.datetime.now() + datetime.timedelta(days=30))
+        date_last = ft.DatePicker(first_date= datetime.datetime.now(),
+                                  last_date= datetime.datetime.now() + datetime.timedelta(days=30))
+        page_.overlay.append(date_first)
+        page_.overlay.append(date_last)
+        date_first_text = ft.Text("")
+        date_last_text = ft.Text("")
 
-        controls_list = []
-        async def get_id(e):
+        async def first_date_text_change(e):
             pass
+
+        async def first_callback(e):
+            await date_first.pick_date_async()
+
+        async def last_callback(e):
+            await date_last.pick_date_async()
+
+        dlg = ft.AlertDialog(
+            modal=True,
+            title=ft.Text("Добавит бронирование"),
+            content=ft.Column(controls=[
+                name_field,
+                surname_field,
+                passport_field,
+                room_id_field,
+                ft.TextButton("Выбрать дату начала", on_click=first_callback),
+                ft.TextButton("Выбрать дату конца", on_click=last_callback)
+
+            ]),
+            actions=[ft.TextButton(text="Добавить", on_click=add), ft.TextButton(text="Отмена", on_click=close)])
+
 
-        class Guest:
-            def __init__(self, id, first_name, last_name):
-                self.id = id
-                self.first_name = first_name
-                self.last_name = last_name
-
-            def get_id(self, e):
-                global guest_id
-                guest_id = self.id
-
-            def get_control(self):
-                return ft.ListTile(title=ft.Text(f"{self.first_name} {self.last_name}"), on_click=self.get_id)
-
-        for i in cursor.execute(f'''SELECT * FROM guests''').fetchall():
-            controls_list.append(Guest(i[0], i[1], i[2]).get_control())
-        async def search(e):
-            global controls_list
-            print(controls_list)
-            new_list = []
-            for i in controls_list:
-                if i.title.value.startswith(guests_selection_dates.value):
-                    new_list.append(i)
-            controls_list = new_list
-            print(controls_list)
-            guests_selection_dates.controls = controls_list
-            await guests_selection_dates.update_async()
-            await page_.update_async()
+        dlg.open = True
+        page_.dialog = dlg
+        await page_.update_async()
 
-        guests_selection_dates = ft.SearchBar(
-            view_elevation=4,
-            divider_color=ft.colors.AMBER,
-            bar_hint_text="Поиск постояльца...",
-            view_hint_text="Выберите постояльца из списка...",
-            # on_change = search,
-            view_leading=ft.IconButton(icon=ft.icons.SEARCH, on_click=search),
-            # on_submit = lambda e: print("submit"),
-            # on_tap= ,
-            controls=controls_list 
-            #    controls=[
-            # ft.ListTile(title=ft.Text(f"Color {i}"), on_click=close_anchor, data=i) for i in range(10)
-
+    # page_.floating_action_button = ft.FloatingActionButton(icon=ft.icons.ADD, on_click=add_res)
 
-        ) 
+    # # async def add_guest(e):
+    # #     async def close_dlg(e):
+    # #         dlg.open = False
+    # #         await page_.update_async()
+
+    # #     async def add_guest_db(e):
+    # #         cursor.execute(f'''INSERT INTO guests(first_name, last_name) VALUES(?, ?)''', (name_field.value, familia_field.value))
+    # #         db.commit()
+    # #         dlg.open = False
+    # #         await page_.update_async()
+
+    # #     name_field = ft.TextField(label="Имя")
+    # #     familia_field = ft.TextField(label="Фамилия")
+
+    # #     dlg = ft.AlertDialog(modal=True, 
+    # #                          title=ft.Text("Добавить Постояльца"),
+    # #                          content=ft.Column(controls=[name_field,
+    # #                                                      familia_field,
+    # #                          ], alignment=ft.alignment.center, horizontal_alignment=ft.CrossAxisAlignment.CENTER,
+    # #                              tight=True),
+    # #                          actions=[ft.TextButton(text="Далее", on_click=add_guest_db), ft.TextButton(text="Отмена", on_click=close_dlg)])
+
+    # #     dlg.open = True
+    # #     page_.dialog = dlg
+    # #     await page_.update_async()
 
-        # rooms_selection = 
+    # async def add_reservation(e):
+    #     # global controls_list
+    #     # guest_id = None
 
+    #     async def close_dlg(e):
+    #         dlg.open = False
+    #         await page_.update_async()
 
-        dlg = ft.AlertDialog(modal=True, 
-                            title=ft.Text("Добавить Постояльца"),
-                            content=ft.Column(controls=[guests_selection_dates]),
-                            actions=[ft.TextButton(text="Далее"), ft.TextButton(text="Отмена", on_click=close_dlg)])
-        page_.dialog = dlg
-        dlg.open = True
-        await page_.update_async()
+    #     # controls_list = []
+
+    #     # class Guest:
+    #     #     def __init__(self, id, first_name, last_name):
+    #     #         self.id = id
+    #     #         self.first_name = first_name
+    #     #         self.last_name = last_name
+
+    #     #     def get_id(self, e):
+    #     #         global guest_id
+    #     #         guest_id = self.id
+
+    #     #     def get_control(self):
+    #     #         return ft.ListTile(title=ft.Text(f"{self.first_name} {self.last_name}"), on_click=self.get_id)
+
+    #     # for i in cursor.execute(f'''SELECT * FROM guests''').fetchall():
+    #     #     controls_list.append(Guest(i[0], i[1], i[2]).get_control())
+    #     # async def search(e):
+    #     #     global controls_list
+    #     #     print(controls_list)
+    #     #     new_list = []
+    #     #     for i in controls_list:
+    #     #         if i.title.value.startswith(guests_selection_dates.value):
+    #     #             new_list.append(i)
+    #     #     controls_list = new_list
+    #     #     print(controls_list)
+    #     #     guests_selection_dates.controls = controls_list
+    #     #     await guests_selection_dates.update_async()
+    #     #     await page_.update_async()
+
+    #     # guests_selection_dates = ft.SearchBar(
+    #     #     view_elevation=4,
+    #     #     divider_color=ft.colors.AMBER,
+    #     #     bar_hint_text="Поиск постояльца...",
+    #     #     view_hint_text="Выберите постояльца из списка...",
+    #     #     # on_change = search,
+    #     #     view_leading=ft.IconButton(icon=ft.icons.SEARCH, on_click=search),
+    #     #     # on_submit = lambda e: print("submit"),
+    #     #     # on_tap= ,
+    #     #     controls=controls_list)
+
+    #     First_Name =  
+
+    #     dlg = ft.AlertDialog(modal=True, 
+    #                         title=ft.Text("Добавить бронирование"),
+    #                         content=ft.Column(controls=[guests_selection_dates]),
+    #                         actions=[ft.TextButton(text="Далее"), ft.TextButton(text="Отмена", on_click=close_dlg)])
+    #     page_.dialog = dlg
+    #     dlg.open = True
+    #     await page_.update_async()
+
+
 
     view_ = ft.View("/main", [ 
-        ft.Row(controls=[ft.FilledButton(text="Добавить Постояльца", on_click=add_guest),
-                         ft.FilledButton(text="Заселить постояльца", on_click=add_reservation),
-                         ft.FilledButton(text="Выселить постояльца"),
-                         ft.FilledButton(text="Свободные номера")
-                         ])
+        ft.Row(controls=[
+
+        ]),
+        table
          ])
+
+    view_.floating_action_button = ft.FloatingActionButton(icon=ft.icons.ADD, on_click=add_res)
 
     view_.appbar = ft.AppBar(leading=ft.Icon(ft.icons.HOTEL), title=ft.Text("Hostel"),
                              bgcolor=ft.colors.SURFACE_VARIANT,