ce884154d6
Adds a grid/thumbnail mode (switchable to the list mode) last mode used is saved faster load and less memory usage add sort by size
123 lines
4.6 KiB
Python
123 lines
4.6 KiB
Python
import logging
|
|
import os
|
|
import gi
|
|
|
|
gi.require_version("Gtk", "3.0")
|
|
|
|
|
|
class KlippyFiles:
|
|
def __init__(self, screen):
|
|
self._screen = screen
|
|
self.callbacks = []
|
|
self.files = {}
|
|
self.directories = []
|
|
self.gcodes_path = None
|
|
|
|
def set_gcodes_path(self):
|
|
virtual_sdcard = self._screen.printer.get_config_section("virtual_sdcard")
|
|
if virtual_sdcard and "path" in virtual_sdcard:
|
|
self.gcodes_path = os.path.expanduser(virtual_sdcard['path'])
|
|
logging.info(f"Gcodes path: {self.gcodes_path}")
|
|
|
|
def _callback(self, result, method, params):
|
|
if "error" in result:
|
|
logging.debug(result["error"])
|
|
return
|
|
if method == "server.files.list":
|
|
for item in result["result"]:
|
|
self.files[item["path"]] = item
|
|
self.request_metadata(item["path"])
|
|
elif method == "server.files.metadata":
|
|
for x in result['result']:
|
|
self.files[params['filename']][x] = result['result'][x]
|
|
if "thumbnails" in self.files[params['filename']]:
|
|
self.files[params['filename']]['thumbnails'].sort(key=lambda y: y['size'], reverse=True)
|
|
for thumbnail in self.files[params['filename']]['thumbnails']:
|
|
thumbnail['local'] = False
|
|
if self.gcodes_path is not None:
|
|
path = os.path.join(
|
|
os.path.dirname(os.path.join(self.gcodes_path, params['filename'])),
|
|
thumbnail['relative_path']
|
|
)
|
|
if os.access(path, os.R_OK):
|
|
thumbnail['local'] = True
|
|
thumbnail['path'] = path
|
|
if thumbnail['local'] is False:
|
|
thumbnail['path'] = os.path.join(
|
|
os.path.dirname(params['filename']),
|
|
thumbnail['relative_path']
|
|
)
|
|
self.run_callbacks("update_metadata", result["result"])
|
|
|
|
def add_file(self, item):
|
|
if 'path' not in item:
|
|
logging.info(f"Error adding item, unknown path: {item}")
|
|
return
|
|
self.files[item['path']] = item
|
|
self.request_metadata(item['path'])
|
|
|
|
def remove_file(self, filename):
|
|
if filename in self.files:
|
|
self.files.pop(filename)
|
|
|
|
def add_callback(self, callback):
|
|
self.callbacks.append(callback)
|
|
|
|
def remove_callback(self, callback):
|
|
if callback in self.callbacks:
|
|
self.callbacks.remove(callback)
|
|
return
|
|
logging.info(f"callback not found {callback}")
|
|
|
|
def process_update(self, data):
|
|
if 'item' in data and data['item']['root'] != 'gcodes':
|
|
return
|
|
|
|
if data['action'] == "create_file":
|
|
self.add_file(data['item'])
|
|
elif data['action'] == "delete_file":
|
|
self.remove_file(data['item']['path'])
|
|
elif data['action'] == "modify_file":
|
|
self.request_metadata(data['item']['path'])
|
|
elif data['action'] == "move_file":
|
|
self.remove_file(data['source_item']['path'])
|
|
self.add_file(data['item'])
|
|
self.run_callbacks(data['action'], data['item'])
|
|
|
|
return False
|
|
|
|
def file_metadata_exists(self, filename):
|
|
return filename in self.files and "slicer" in self.files[filename]
|
|
|
|
def get_thumbnail_location(self, filename, small=False):
|
|
if all((
|
|
small,
|
|
len(self.files[filename]['thumbnails']) > 1
|
|
)):
|
|
thumb = self.files[filename]['thumbnails'][1]
|
|
else:
|
|
thumb = self.files[filename]['thumbnails'][0]
|
|
return ['file', thumb['path']] if thumb['local'] else ['http', thumb['path']]
|
|
|
|
def has_thumbnail(self, filename):
|
|
return filename in self.files and "thumbnails" in self.files[filename]
|
|
|
|
def request_metadata(self, filename):
|
|
if os.path.splitext(filename)[1] in {'.gcode', '.gco', '.g'}:
|
|
self._screen._ws.klippy.get_file_metadata(filename, self._callback)
|
|
|
|
def refresh_files(self):
|
|
self._screen._ws.klippy.get_file_list(self._callback)
|
|
|
|
def run_callbacks(self, action, item):
|
|
for cb in self.callbacks:
|
|
cb(action, item)
|
|
|
|
def get_file_info(self, filename):
|
|
if filename not in self.files:
|
|
return {"path": None, "modified": 0, "size": 0}
|
|
return self.files[filename]
|
|
|
|
def get_dir_info(self, directory):
|
|
self._screen._ws.klippy.get_dir_info(self._callback, directory=directory)
|