remoterf 0.0.7.41__py3-none-any.whl

remoteRF/common/__init__.py

@@ -0,0 +1,2 @@
+from .grpc import *
+from .utils import *

remoteRF/common/grpc/__init__.py

@@ -0,0 +1 @@
+from . import grpc_pb2, grpc_pb2_grpc

remoteRF/common/grpc/grpc_pb2.py

@@ -0,0 +1,59 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: grpc.proto
+# Protobuf Python Version: 5.29.0
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    29,
+    0,
+    '',
+    'grpc.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\ngrpc.proto\x12\tremote_rf\"\xa2\x01\n\x11GenericRPCRequest\x12\x15\n\rfunction_name\x18\x01 \x01(\t\x12\x34\n\x04\x61rgs\x18\x02 \x03(\x0b\x32&.remote_rf.GenericRPCRequest.ArgsEntry\x1a@\n\tArgsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\"\n\x05value\x18\x02 \x01(\x0b\x32\x13.remote_rf.Argument:\x02\x38\x01\"\x96\x01\n\x12GenericRPCResponse\x12;\n\x07results\x18\x01 \x03(\x0b\x32*.remote_rf.GenericRPCResponse.ResultsEntry\x1a\x43\n\x0cResultsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\"\n\x05value\x18\x02 \x01(\x0b\x32\x13.remote_rf.Argument:\x02\x38\x01\"\x19\n\nArrayShape\x12\x0b\n\x03\x64im\x18\x01 \x03(\x05\"+\n\rComplexNumber\x12\x0c\n\x04real\x18\x01 \x01(\x02\x12\x0c\n\x04imag\x18\x02 \x01(\x02\"a\n\x11\x43omplexNumpyArray\x12$\n\x05shape\x18\x01 \x01(\x0b\x32\x15.remote_rf.ArrayShape\x12&\n\x04\x64\x61ta\x18\x02 \x03(\x0b\x32\x18.remote_rf.ComplexNumber\"D\n\x0eRealNumpyArray\x12$\n\x05shape\x18\x01 \x01(\x0b\x32\x15.remote_rf.ArrayShape\x12\x0c\n\x04\x64\x61ta\x18\x02 \x03(\x02\"\xd7\x01\n\x08\x41rgument\x12\x16\n\x0cstring_value\x18\x01 \x01(\tH\x00\x12\x15\n\x0bint64_value\x18\x02 \x01(\x03H\x00\x12\x15\n\x0b\x66loat_value\x18\x03 \x01(\x02H\x00\x12\x14\n\nbool_value\x18\x04 \x01(\x08H\x00\x12\x35\n\rcomplex_array\x18\x05 \x01(\x0b\x32\x1c.remote_rf.ComplexNumpyArrayH\x00\x12/\n\nreal_array\x18\x06 \x01(\x0b\x32\x19.remote_rf.RealNumpyArrayH\x00\x42\x07\n\x05value2Q\n\nGenericRPC\x12\x43\n\x04\x43\x61ll\x12\x1c.remote_rf.GenericRPCRequest\x1a\x1d.remote_rf.GenericRPCResponseB\x1e\n\x10\x63om.example.demoB\nDemoProtosb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'grpc_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\020com.example.demoB\nDemoProtos'
+  _globals['_GENERICRPCREQUEST_ARGSENTRY']._loaded_options = None
+  _globals['_GENERICRPCREQUEST_ARGSENTRY']._serialized_options = b'8\001'
+  _globals['_GENERICRPCRESPONSE_RESULTSENTRY']._loaded_options = None
+  _globals['_GENERICRPCRESPONSE_RESULTSENTRY']._serialized_options = b'8\001'
+  _globals['_GENERICRPCREQUEST']._serialized_start=26
+  _globals['_GENERICRPCREQUEST']._serialized_end=188
+  _globals['_GENERICRPCREQUEST_ARGSENTRY']._serialized_start=124
+  _globals['_GENERICRPCREQUEST_ARGSENTRY']._serialized_end=188
+  _globals['_GENERICRPCRESPONSE']._serialized_start=191
+  _globals['_GENERICRPCRESPONSE']._serialized_end=341
+  _globals['_GENERICRPCRESPONSE_RESULTSENTRY']._serialized_start=274
+  _globals['_GENERICRPCRESPONSE_RESULTSENTRY']._serialized_end=341
+  _globals['_ARRAYSHAPE']._serialized_start=343
+  _globals['_ARRAYSHAPE']._serialized_end=368
+  _globals['_COMPLEXNUMBER']._serialized_start=370
+  _globals['_COMPLEXNUMBER']._serialized_end=413
+  _globals['_COMPLEXNUMPYARRAY']._serialized_start=415
+  _globals['_COMPLEXNUMPYARRAY']._serialized_end=512
+  _globals['_REALNUMPYARRAY']._serialized_start=514
+  _globals['_REALNUMPYARRAY']._serialized_end=582
+  _globals['_ARGUMENT']._serialized_start=585
+  _globals['_ARGUMENT']._serialized_end=800
+  _globals['_GENERICRPC']._serialized_start=802
+  _globals['_GENERICRPC']._serialized_end=883
+# @@protoc_insertion_point(module_scope)

remoteRF/common/grpc/grpc_pb2_grpc.py

@@ -0,0 +1,97 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+from . import grpc_pb2 as grpc__pb2
+
+GRPC_GENERATED_VERSION = '1.71.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in grpc_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )
+
+
+class GenericRPCStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.Call = channel.unary_unary(
+                '/remote_rf.GenericRPC/Call',
+                request_serializer=grpc__pb2.GenericRPCRequest.SerializeToString,
+                response_deserializer=grpc__pb2.GenericRPCResponse.FromString,
+                _registered_method=True)
+
+
+class GenericRPCServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def Call(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_GenericRPCServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'Call': grpc.unary_unary_rpc_method_handler(
+                    servicer.Call,
+                    request_deserializer=grpc__pb2.GenericRPCRequest.FromString,
+                    response_serializer=grpc__pb2.GenericRPCResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'remote_rf.GenericRPC', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('remote_rf.GenericRPC', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class GenericRPC(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def Call(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/remote_rf.GenericRPC/Call',
+            grpc__pb2.GenericRPCRequest.SerializeToString,
+            grpc__pb2.GenericRPCResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

remoteRF/common/utils/__init__.py

@@ -0,0 +1,4 @@
+from .api_token import validate_token, generate_token, hash_token
+from .process_arg import unmap_arg, map_arg
+from .ansi_codes import printf, stylize, Sty
+from .list_string import list_to_str, str_to_list

remoteRF/common/utils/ansi_codes.py

@@ -0,0 +1,120 @@
+from prompt_toolkit.styles import Style
+from prompt_toolkit.formatted_text import FormattedText
+from prompt_toolkit import print_formatted_text
+from enum import Enum
+
+class Sty(Enum):
+    # Basic colors
+    RED = 'red'
+    GREEN = 'green'
+    BLUE = 'blue'
+    YELLOW = 'yellow'
+    MAGENTA = 'magenta'
+    CYAN = 'cyan'
+    GRAY = 'gray'
+    
+    # Background colors
+    BG_RED = 'bg-red'
+    BG_GREEN = 'bg-green'
+    BG_BLUE = 'bg-blue'
+    
+    # Bright versions
+    BRIGHT_RED = 'bright-red'
+    BRIGHT_GREEN = 'bright-green'
+    BRIGHT_BLUE = 'bright-blue'
+    
+    # Formatting
+    BOLD = 'bold'
+    ITALIC = 'italic'
+    UNDERLINE = 'underline'
+    BLINK = 'blink'
+    REVERSE = 'reverse'
+    
+    # Combinations
+    ERROR = 'error'
+    WARNING = 'warning'
+    INFO = 'info'
+    
+    # Special
+    SELECTED = 'selected'
+    DEFAULT = 'default'
+
+# Define the styles based on ANSI codes
+style = Style.from_dict({
+    # Basic colors
+    'red': 'fg:#110000',
+    'green': 'fg:#003300',
+    'blue': 'fg:#0000ff',
+    'yellow': 'fg:#ffff00',
+    'magenta': 'fg:#ff00ff',
+    'cyan': 'fg:#00ffff',
+    'gray': 'fg:#808080',
+    
+    # Bright versions
+    'bright-red': 'fg:#ff5555',
+    'bright-green': 'fg:#00ff00',
+    'bright-blue': 'fg:#5555ff',
+    
+    # Formatting
+    'bold': 'bold',
+    'italic': 'italic',
+    'underline': 'underline',
+    'reverse': 'reverse',
+    
+    # Combinations
+    'error': 'bg:#ff0000 fg:#ffffff bold',
+    'warning': 'bg:#ffff00 fg:#000000 bold',
+    'info': 'bg:#0000ff fg:#ffffff italic underline',
+    
+    # Special
+    'selected': 'bg:#ffffff #000000 reverse',
+    'default':''
+})
+
+def printf(*args) -> str:
+    if len(args) % 2 != 0:
+        raise ValueError('Arguments must be in pairs of two.')
+    
+    # Create formatted text using the defined style
+    formatted_text = []
+    
+    for i in range(0, len(args), 2):
+        message = args[i]
+        styles = args[i+1]
+        
+        if not isinstance(styles, tuple):
+            styles = (styles,)
+            
+        resolved_styles = (s.value if isinstance(s, Enum) else s for s in styles)
+        
+        style_class = ' '.join(resolved_styles)
+        
+        formatted_text.append(('class:' + style_class, message))
+    
+    # Create FormattedText object from pairs
+    text = FormattedText(formatted_text)
+    
+    print_formatted_text(text, style=style)
+    return text
+
+def stylize(*args):
+    """
+    Create a styled prompt text based on pairs of (text, (Sty, ...), ...).
+    """
+    if len(args) % 2 != 0:
+        raise ValueError("Arguments must be in pairs of (text, style_class).")
+    
+    styled_parts = []
+    for i in range(0, len(args), 2):
+        text = args[i]
+        styles = args[i + 1]
+        
+        if not isinstance(styles, tuple):
+            styles = (styles,)
+            
+        resolved_styles = (s.value if isinstance(s, Enum) else s for s in styles)
+        
+        style_class = ' '.join(resolved_styles)
+        styled_parts.append(('class:' + style_class, text))
+    
+    return FormattedText(styled_parts)

remoteRF/common/utils/api_token.py

@@ -0,0 +1,31 @@
+import os
+import hashlib
+import base64
+import secrets
+from dotenv import load_dotenv, find_dotenv
+
+### API Token Management
+# API Tokens are used to authenticate clients to the device
+# API Tokens are stored locally on the device in its .env file
+
+def generate_token(length=8) -> tuple[str, str, str]:
+    random_bytes = secrets.token_bytes(length)  # Generate a random byte string
+    token = base64.urlsafe_b64encode(random_bytes).decode('utf-8').rstrip('=')  # Encode the byte string in a URL-safe base64 format
+    salt = os.urandom(16).hex()  # 16 bytes of random salt
+    hashed = hashlib.sha256(bytes.fromhex(salt) + token.encode()).hexdigest()  # Hash to sha256 standard
+    return salt, hashed, token
+
+def validate_token(salt, hash, token) -> bool:
+    new_hashed = hashlib.sha256(bytes.fromhex(salt) + token.encode()).hexdigest()
+    return new_hashed == hash
+
+def hash_token(token: str) -> tuple[str, str]:
+    salt = os.urandom(16).hex()
+    hashed = hashlib.sha256(bytes.fromhex(salt) + token.encode()).hexdigest()
+    return salt, hashed
+
+# Example Usage:
+# if __name__ == '__main__':
+#     okay = generate_token()
+#     assert validate_token(okay[0], okay[1], okay[2] + "s"), "Token validation failed!"
+#     print("Token validation succeeded!")

remoteRF/common/utils/list_string.py

@@ -0,0 +1,5 @@
+def list_to_str(list:list) -> str:
+    return ','.join(str(x) for x in list)
+
+def str_to_list(s:str) -> list[int]:
+    return [int(x) for x in s.split(',')]

remoteRF/common/utils/process_arg.py

@@ -0,0 +1,80 @@
+from ..grpc import grpc_pb2, grpc_pb2_grpc
+import numpy as np
+
+def unmap_arg(arg):
+    if arg.HasField('int64_value'):
+        return arg.int64_value
+    elif arg.HasField('float_value'):
+        return arg.float_value
+    elif arg.HasField('string_value'):
+        return arg.string_value
+    elif arg.HasField('bool_value'):
+        return arg.bool_value
+    elif arg.HasField('real_array'):
+        shape = tuple(arg.real_array.shape.dim)
+        return np.array(arg.real_array.data, dtype=np.float64).reshape(shape)
+    elif arg.HasField('complex_array'):
+        shape = tuple(arg.complex_array.shape.dim)
+        data = [complex(c.real, c.imag) for c in arg.complex_array.data]
+        return np.array(data, dtype=np.complex64).reshape(shape)
+    else:
+        raise ValueError(f"Unknown argument type during unmapping: {arg}")
+    
+def map_arg(value):
+    arg = grpc_pb2.Argument()
+    
+    if isinstance(value, int):
+        arg.int64_value = value
+    elif isinstance(value, float):
+        arg.float_value = value
+    elif isinstance(value, str):
+        arg.string_value = value
+    elif isinstance(value, bool):
+        arg.bool_value = value
+    elif isinstance(value, np.ndarray):
+        if np.iscomplexobj(value):
+            complex_array = arg.complex_array
+            complex_array.shape.dim.extend(value.shape)
+            for num in value.ravel():
+                complex_num = complex_array.data.add()
+                complex_num.real = num.real
+                complex_num.imag = num.imag
+        else:
+            float_array = arg.real_array
+            float_array.shape.dim.extend(value.shape)
+            float_array.data.extend(value.ravel())
+    else:
+        raise ValueError(f"Unknown argument type during mapping: {value}")
+    return arg
+        
+def map_array_proto(np_array):
+    arg = grpc_pb2.Argument()
+    
+    # Check if the array is complex
+    if np.iscomplexobj(np_array):
+        complex_array = grpc_pb2.ComplexArray()
+        for num in np_array.flat:
+            complex_number = complex_array.data.add()
+            complex_number.real = num.real
+            complex_number.imag = num.imag
+        arg.complex_array.CopyFrom(complex_array)
+    else:
+        # Handle as a regular float array
+        float_array = grpc_pb2.FloatArray()
+        float_array.data.extend(np_array.flat)
+        arg.float_array.CopyFrom(float_array)
+
+    return arg
+
+def unmap_array_proto(arg):
+    # Check which type of array is available and convert appropriately
+    if arg.HasField('complex_array'):
+        # Convert ComplexArray to a numpy array of complex numbers
+        data = [complex(cn.real, cn.imag) for cn in arg.complex_array.data]
+        return np.array(data, dtype=np.complex64)
+    elif arg.HasField('float_array'):
+        # Convert FloatArray to a numpy array of floats
+        return np.array(arg.float_array.data, dtype=np.float32)
+    else:
+        raise ValueError("Argument does not contain a recognizable array.")
+

remoteRF/core/__init__.py

@@ -0,0 +1,2 @@
+from .grpc_client import rpc_client
+from .grpc_acc import RemoteRFAccount

remoteRF/core/acc_login.py

@@ -0,0 +1,4 @@
+from . import app as client
+
+def main():
+    pass