Module `refinery.lib.lnk.stringdata`

Expand source code Browse git

from __future__ import annotations

import codecs

from dataclasses import dataclass

from refinery.lib.lnk.flags import LinkFlags
from refinery.lib.structures import StructReader


@dataclass
class StringData:
    description: str | None = None
    relative_path: str | None = None
    working_directory: str | None = None
    command_line_arguments: str | None = None
    icon_location: str | None = None

    def __json__(self) -> dict:
        return {k: v for k, v in self.__dict__.items() if v is not None}

    @classmethod
    def parse(
        cls,
        reader: StructReader[memoryview],
        flags: LinkFlags,
    ) -> StringData:
        result = cls()
        is_unicode = flags.IsUnicode
        entries = (
            (LinkFlags.HasName, 'description'),
            (LinkFlags.HasRelativePath, 'relative_path'),
            (LinkFlags.HasWorkingDir, 'working_directory'),
            (LinkFlags.HasArguments, 'command_line_arguments'),
            (LinkFlags.HasIconLocation, 'icon_location'),
        )
        for flag, attr in entries:
            if not flags & flag:
                continue
            count = reader.u16()
            if is_unicode:
                raw = reader.read(count * 2)
                value = codecs.decode(raw, 'utf-16-le')
            else:
                raw = reader.read(count)
                try:
                    value = codecs.decode(raw, 'cp1252')
                except Exception:
                    value = bytes(raw).hex()
            setattr(result, attr, value)
        return result

Classes

class StringData (description=None, relative_path=None, working_directory=None, command_line_arguments=None, icon_location=None)

Expand source code Browse git

@dataclass
class StringData:
    description: str | None = None
    relative_path: str | None = None
    working_directory: str | None = None
    command_line_arguments: str | None = None
    icon_location: str | None = None

    def __json__(self) -> dict:
        return {k: v for k, v in self.__dict__.items() if v is not None}

    @classmethod
    def parse(
        cls,
        reader: StructReader[memoryview],
        flags: LinkFlags,
    ) -> StringData:
        result = cls()
        is_unicode = flags.IsUnicode
        entries = (
            (LinkFlags.HasName, 'description'),
            (LinkFlags.HasRelativePath, 'relative_path'),
            (LinkFlags.HasWorkingDir, 'working_directory'),
            (LinkFlags.HasArguments, 'command_line_arguments'),
            (LinkFlags.HasIconLocation, 'icon_location'),
        )
        for flag, attr in entries:
            if not flags & flag:
                continue
            count = reader.u16()
            if is_unicode:
                raw = reader.read(count * 2)
                value = codecs.decode(raw, 'utf-16-le')
            else:
                raw = reader.read(count)
                try:
                    value = codecs.decode(raw, 'cp1252')
                except Exception:
                    value = bytes(raw).hex()
            setattr(result, attr, value)
        return result

Static methods

def parse(reader, flags)

Instance variables

var description: The type of the None singleton.
var relative_path: The type of the None singleton.
var working_directory: The type of the None singleton.
var command_line_arguments: The type of the None singleton.
var icon_location: The type of the None singleton.