33from sqlmodel import select
44import reflex as rx
55
6- from data_visualisation .models import Customer , Cereals , Covid , Countries
6+ from data_visualisation .models import Customer , Cereals , Covid , Countries # noqa: F401
77from data_visualisation .data_loading import loading_data
88
99
@@ -57,7 +57,7 @@ def add_item(self):
5757 session .add (MODEL (** self .current_item ))
5858 session .commit ()
5959 self .load_entries ()
60- return rx .window_alert (f "Item has been added." )
60+ return rx .window_alert ("Item has been added." )
6161
6262 def update_item (self ):
6363 """Update an item in the database."""
@@ -66,7 +66,7 @@ def update_item(self):
6666 select (MODEL ).where (MODEL .id == self .current_item ["id" ])
6767 ).first ()
6868
69- for field in MODEL .get_fields () :
69+ for field in MODEL .__fields__ :
7070 if field != "id" :
7171 setattr (item , field , self .current_item [field ])
7272 session .add (item )
@@ -156,11 +156,7 @@ def add_item_ui():
156156 ),
157157 rx .form (
158158 rx .flex (
159- * [
160- add_fields (field )
161- for field in MODEL .get_fields ()
162- if field != "id"
163- ],
159+ * [add_fields (field ) for field in MODEL .__fields__ if field != "id" ],
164160 rx .box (
165161 rx .button (
166162 "Submit" ,
@@ -225,7 +221,7 @@ def update_item_ui(item):
225221 update_fields_and_attrs (
226222 field , getattr (State .current_item , field )
227223 )
228- for field in MODEL .get_fields ()
224+ for field in MODEL .__fields__
229225 if field != "id"
230226 ],
231227 rx .box (
@@ -294,7 +290,7 @@ def show_item(item: MODEL):
294290 rx .table .cell (rx .avatar (fallback = "DA" )),
295291 * [
296292 rx .table .cell (getattr (item , field ))
297- for field in MODEL .get_fields ()
293+ for field in MODEL .__fields__
298294 if field != "id"
299295 ],
300296 rx .table .cell (
@@ -323,7 +319,7 @@ def content():
323319 ),
324320 rx .spacer (),
325321 rx .select (
326- [* [field for field in MODEL .get_fields () if field != "id" ]],
322+ [* [field for field in MODEL .__fields__ if field != "id" ]],
327323 placeholder = "Sort By: Name" ,
328324 size = "3" ,
329325 on_change = lambda sort_value : State .sort_values (sort_value ),
@@ -340,7 +336,7 @@ def content():
340336 rx .table .column_header_cell ("Icon" ),
341337 * [
342338 rx .table .column_header_cell (field )
343- for field in MODEL .get_fields ()
339+ for field in MODEL .__fields__
344340 if field != "id"
345341 ],
346342 rx .table .column_header_cell ("Edit" ),
0 commit comments