Merge pull request 'clan_cli: ClanURI fixed non executing tests' (#611) from Qubasa-main into main

This commit is contained in:
clan-bot
2023-12-06 12:41:16 +00:00
3 changed files with 31 additions and 33 deletions

View File

@@ -4,7 +4,7 @@ import urllib.parse
from dataclasses import dataclass
from enum import Enum, member
from pathlib import Path
from typing import Dict
from typing import Dict, Self
from .errors import ClanError
@@ -41,8 +41,7 @@ class ClanScheme(Enum):
# so make sure there are no conflicts with other webservices
@dataclass
class ClanParameters:
flake_attr: str | None
machine: str | None
flake_attr: str = "defaultVM"
# Define the ClanURI class
@@ -60,30 +59,25 @@ class ClanURI:
self._components = urllib.parse.urlparse(self._nested_uri)
# Parse the query string into a dictionary
self._query = urllib.parse.parse_qs(self._components.query)
query = urllib.parse.parse_qs(self._components.query)
params: Dict[str, str | None] = {}
params: Dict[str, str] = {}
for field in dataclasses.fields(ClanParameters):
if field.name in self._query:
# Check if the field type is a list
if issubclass(field.type, list):
setattr(params, field.name, self._query[field.name])
# Check if the field type is a single value
else:
values = self._query[field.name]
if len(values) > 1:
raise ClanError(
"Multiple values for parameter: {}".format(field.name)
)
setattr(params, field.name, values[0])
if field.name in query:
values = query[field.name]
if len(values) > 1:
raise ClanError(
"Multiple values for parameter: {}".format(field.name)
)
params[field.name] = values[0]
# Remove the field from the query dictionary
# clan uri and nested uri share one namespace for query parameters
# we need to make sure there are no conflicts
del self._query[field.name]
else:
params[field.name] = None
del query[field.name]
new_query = urllib.parse.urlencode(query, doseq=True)
self._components = self._components._replace(query=new_query)
self.params = ClanParameters(**params)
# Use the match statement to check the scheme and create a ClanScheme member with the value
@@ -98,3 +92,9 @@ class ClanURI:
raise ClanError(
"Unsupported scheme: {}".format(self._components.scheme)
)
@classmethod
def from_path(cls, path: Path, params: ClanParameters) -> Self: # noqa
urlparams = urllib.parse.urlencode(params.__dict__)
return cls("clan://file://{}?{}".format(path, urlparams))