utils.py

import sys

from kivy.config import Config
from kivy.effects.scroll import ScrollEffect

from kivymd.uix.boxlayout import MDBoxLayout
from kivymd.uix.label import MDLabel
from kivymd.uix.snackbar import MDSnackbar, MDSnackbarText


def create_snackbar(message='Default'):
    t = MDSnackbarText(
        size_hint=(None, None),
        halign='center', valign='middle',
        text=message, text_size=(None, None),
        allow_copy=True
    )

    t.texture_update()
    t.width = t.texture_size[0]

    snackbar = MDSnackbar(
        t,
        y=24,
        adaptive_width=True,
        pos_hint={'center_x': 0.5},
        duration=1.2 + int(len(message) * 0.01),
        padding=[15, 8, 15, 8]
    )

    snackbar.ids.label_container.padding = [0, 0, 0, 0]
    snackbar.ids.label_container.size_hint_x = None
    snackbar.ids.label_container.width = snackbar.ids.label_container.children[0].width

    snackbar.size_hint_x = None
    snackbar.width = snackbar.children[-1].width + 30  # Account for horizontal padding

    return snackbar


def reset_defaults(app):
    """ Reset kivy config (Debugging) """
    Config._sections.clear()

    # Not sure if this actually works:

    # for section in Config.sections():
    #     for option in Config.options(section):
    #         Config.setdefault(section, option, Config.get(section, option))

    Config.write()
    app.get_running_app().stop()  # And restart


def convert_bytes(size):
    """Convert bytes to a human-readable format (KB, MB, GB)."""
    for unit in ['bytes', 'KB', 'MB', 'GB', 'TB']:
        if size < 1024.0:
            return f'{size:.2f} {unit}'
        size /= 1024.0


class NoOverscrollEffect(ScrollEffect):
    """Custom Scroll Effect to Disable Overscroll.
    Used by the navigation drawer and scroll view widgets."""

    def convert_overscroll(self, *args):
        # Override to prevent overscroll
        return 0, 0  # No overscroll effect

    def reset_scale(self, *args):
        # Override to prevent overscroll
        return 0, 0  # No overscroll effect

    def on_scroll_stop(self):
        # Prevent overscroll by clamping scroll values
        self.scroll_y = max(0, min(self.scroll_y, 1))
        self.scroll_x = max(0, min(self.scroll_x, 1))


def get_size(obj, seen=None):
    """ Recursively finds size of objects """
    size = sys.getsizeof(obj)
    if seen is None:
        seen = set()
    obj_id = id(obj)
    if obj_id in seen:
        return 0
    # Important mark as seen *before* entering recursion to gracefully handle
    # self-referential objects
    seen.add(obj_id)
    if isinstance(obj, dict):
        size += sum([get_size(v, seen) for v in obj.values()])
        size += sum([get_size(k, seen) for k in obj.keys()])
    elif hasattr(obj, '__dict__'):
        size += get_size(obj.__dict__, seen)
    elif hasattr(obj, '__iter__') and not isinstance(obj, (str, bytes, bytearray)):
        size += sum([get_size(i, seen) for i in obj])
    return size


def callback_update_label_size(instance, value, label_sv=None):
    instance.text_size = [None, None]
    instance.texture_update()
    if label_sv:
        w, h = instance.size = instance.texture_size
        label_sv.size = (min(w, 350), min(h, 350))


def generate_message_cont(text='', padding=(13, 13, 13, 13)):
    """ Dialog content generating helper function """
    from widgets.scrollview import ZScrollView

    a_label = MDLabel(
        size_hint=(None, None),
        halign='center', valign='middle',
        font_style='Table Cell Mono',
        text=text, text_size=(None, None),
        padding=padding,
        allow_selection=True,
        allow_copy=True
    )

    label_sv = ZScrollView(do_scroll_x=True)
    label_sv.add_widget(a_label)
    label_sv.size_hint = (None, None)
    a_label.bind(text=lambda *args: callback_update_label_size(*args, label_sv=label_sv))

    container = MDBoxLayout(label_sv,
                            orientation='vertical',
                            adaptive_size=True,
                            )
    return container


def set_and_dismiss(objects, attributes, values, obj_to_dismiss=None):
    """ Apply changes to widgets & dismiss dialog """
    for obj, attribute, value in zip(objects, attributes, values):
        setattr(obj, attribute, value)

    # Dynamically dismiss object
    if obj_to_dismiss and hasattr(obj_to_dismiss, 'dismiss'):
        dismiss_method = getattr(obj_to_dismiss, 'dismiss')
        if callable(dismiss_method):
            dismiss_method()


def numpy_to_dict(array):
    """ Serialize to a JSON-compatible format """
    return {
        'data': array.tolist(),  # Convert to list
        'dtype': str(array.dtype),  # Store data type
        'shape': array.shape  # Store shape
    }


def dict_to_numpy(d):
    """ Deserialize back to NumPy array """
    import numpy as np
    return np.array(d['data'], dtype=d['dtype']).reshape(d['shape'])