@@ -54,14 +54,12 @@ def __init__(
5454
5555 @property
5656 def file_list (self ):
57- """Get the current list of files """
57+ """Available files list"""
5858 return self .files
5959
6060 def handle_message_received (self , method : str , data , params : dict ) -> None :
6161 """Handle file related messages received by moonraker"""
6262 if "server.files.list" in method :
63- # Get all files in root and its subdirectories and
64- # request their metadata
6563 self .files .clear ()
6664 self .files = data
6765 [self .request_file_metadata .emit (item ["path" ]) for item in self .files ]
@@ -73,8 +71,6 @@ def handle_message_received(self, method: str, data, params: dict) -> None:
7371 else :
7472 self .files_metadata [data ["filename" ]] = data
7573 elif "server.files.get_directory" in method :
76- # Emit here the files for each directory so the
77- # ui can build the files list
7874 self .directories = data .get ("dirs" , {})
7975 self .files .clear ()
8076 self .files = data .get ("files" , [])
@@ -99,7 +95,7 @@ def on_request_fileinfo(self, filename: str) -> None:
9995 """Requests metadata for a file
10096
10197 Args:
102- filename (str): file
98+ filename (str): file to get metadata from
10399 """
104100 _data : dict = {
105101 "thumbnail_images" : list ,
@@ -134,7 +130,6 @@ def on_request_fileinfo(self, filename: str) -> None:
134130 )
135131 _thumbnail_images = list (map (lambda path : QtGui .QImage (path ), _thumbnail_paths ))
136132 _data .update ({"thumbnail_images" : _thumbnail_images })
137-
138133 _data .update ({"filament_total" : _file_metadata .get ("filament_total" , "?" )})
139134 _data .update ({"estimated_time" : _file_metadata .get ("estimated_time" , 0 )})
140135 _data .update ({"layer_count" : _file_metadata .get ("layer_count" , - 1.0 )})
@@ -165,18 +160,15 @@ def on_request_fileinfo(self, filename: str) -> None:
165160 self .fileinfo .emit (_data )
166161
167162 def eventFilter (self , a0 : QtCore .QObject , a1 : QtCore .QEvent ) -> bool :
168- """Filter Klippy related events"""
163+ """Handle Websocket and Klippy events"""
164+ if a1 .type () == events .WebSocketOpen .type ():
165+ self .request_file_list .emit ()
166+ self .request_dir_info [str , bool ].emit ("" , False )
167+ return False
169168 if a1 .type () == events .KlippyDisconnected .type ():
170169 self .files_metadata .clear ()
171170 self .files .clear ()
172171 return False
173- if a1 .type () == events .KlippyReady .type ():
174- # Request all files including in subdirectories
175- # in order to get all metadata
176- self .request_file_list .emit ()
177- # List and directory build is depended only on this signal
178- self .request_dir_info [str , bool ].emit ("" , False )
179- return False
180172 return super ().eventFilter (a0 , a1 )
181173
182174 def event (self , a0 : QtCore .QEvent ) -> bool :
0 commit comments