mirror of
https://github.com/arcan1s/ahriman.git
synced 2025-04-24 07:17:17 +00:00
fix: correctly serialize patches from database (#137)
If value is stored as array in database it is serialized as json, but read as normal string, which lead to innability to use list patches This fix also removes any postprocessing (unquoting) for functions
This commit is contained in:
parent
0cc35e70e3
commit
f5d7085325
@ -198,8 +198,7 @@ class Patch(Handler):
|
||||
else:
|
||||
patch = patch_path.read_text(encoding="utf8")
|
||||
# remove spaces around the patch and parse to correct type
|
||||
parsed = PkgbuildPatch.parse(patch.strip())
|
||||
return PkgbuildPatch(variable, parsed)
|
||||
return PkgbuildPatch.parse(variable, patch.strip())
|
||||
|
||||
@staticmethod
|
||||
def patch_set_create(application: Application, package_base: str, patch: PkgbuildPatch) -> None:
|
||||
|
@ -83,7 +83,7 @@ class PatchOperations(Operations):
|
||||
"""
|
||||
def run(connection: Connection) -> list[tuple[str, PkgbuildPatch]]:
|
||||
return [
|
||||
(row["package_base"], PkgbuildPatch(row["variable"], row["patch"]))
|
||||
(row["package_base"], PkgbuildPatch.parse(row["variable"], row["patch"]))
|
||||
for row in connection.execute(
|
||||
"""select * from patches where :package_base is null or package_base = :package_base""",
|
||||
{"package_base": package_base})
|
||||
|
@ -17,6 +17,7 @@
|
||||
# You should have received a copy of the GNU General Public License
|
||||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
#
|
||||
import json
|
||||
import shlex
|
||||
|
||||
from dataclasses import dataclass, fields
|
||||
@ -79,11 +80,11 @@ class PkgbuildPatch:
|
||||
variable(str): variable in bash form, i.e. KEY=VALUE
|
||||
|
||||
Returns:
|
||||
Self: package properties
|
||||
Self: patch object
|
||||
"""
|
||||
key, *value_parts = variable.split("=", maxsplit=1)
|
||||
raw_value = next(iter(value_parts), "") # extract raw value
|
||||
return cls(key, cls.parse(raw_value))
|
||||
return cls.parse(key, raw_value)
|
||||
|
||||
@classmethod
|
||||
def from_json(cls, dump: dict[str, Any]) -> Self:
|
||||
@ -100,21 +101,36 @@ class PkgbuildPatch:
|
||||
known_fields = [pair.name for pair in fields(cls)]
|
||||
return cls(**filter_json(dump, known_fields))
|
||||
|
||||
@staticmethod
|
||||
def parse(source: str) -> str | list[str]:
|
||||
@classmethod
|
||||
def parse(cls, key: str | None, source: str) -> Self:
|
||||
"""
|
||||
parse string value to the PKGBUILD patch value. This method simply takes string, tries to identify it as array
|
||||
or just string and return the respective value. Functions should be processed correctly, however, not guaranteed
|
||||
or just string and return the respective value. Functions are returned as is. Shell arrays and single values
|
||||
are returned without quotes
|
||||
|
||||
Args:
|
||||
source(str): source string to parse
|
||||
key(str | None): variable key
|
||||
source(str): source value string to parse
|
||||
|
||||
Returns:
|
||||
str | list[str]: parsed value either string or list of strings
|
||||
Self: parsed patch object
|
||||
"""
|
||||
if source.startswith("(") and source.endswith(")"):
|
||||
return shlex.split(source[1:-1]) # arrays for poor
|
||||
return PkgbuildPatch.unquote(source)
|
||||
def value() -> str | list[str]:
|
||||
match source:
|
||||
case function if key is not None and key.endswith("()"):
|
||||
# the key looks like a function, no further processing should be applied here
|
||||
return function
|
||||
case shell_array if shell_array.startswith("(") and shell_array.endswith(")"):
|
||||
# the source value looks like shell array, remove brackets and parse with shlex
|
||||
return shlex.split(shell_array[1:-1])
|
||||
case json_array if json_array.startswith("[") and json_array.endswith("]"):
|
||||
# json (aka python) array, parse with json parser instead
|
||||
parsed: list[str] = json.loads(json_array)
|
||||
return parsed
|
||||
case variable:
|
||||
return cls.unquote(variable)
|
||||
|
||||
return cls(key, value())
|
||||
|
||||
@staticmethod
|
||||
def unquote(source: str) -> str:
|
||||
|
@ -15,6 +15,22 @@ def test_patches_get_insert(database: SQLite, package_ahriman: Package, package_
|
||||
]
|
||||
|
||||
|
||||
def test_patches_get_insert_array(database: SQLite, package_ahriman: Package) -> None:
|
||||
"""
|
||||
must insert array patch to database
|
||||
"""
|
||||
database.patches_insert(package_ahriman.base, [PkgbuildPatch("array", ["array", "value"])])
|
||||
assert database.patches_get(package_ahriman.base) == [PkgbuildPatch("array", ["array", "value"])]
|
||||
|
||||
|
||||
def test_patches_get_insert_function(database: SQLite, package_ahriman: Package) -> None:
|
||||
"""
|
||||
must insert function patch to database
|
||||
"""
|
||||
database.patches_insert(package_ahriman.base, [PkgbuildPatch("function()", "{ function body' }")])
|
||||
assert database.patches_get(package_ahriman.base) == [PkgbuildPatch("function()", "{ function body' }")]
|
||||
|
||||
|
||||
def test_patches_list(database: SQLite, package_ahriman: Package, package_python_schedule: Package) -> None:
|
||||
"""
|
||||
must list all patches
|
||||
|
@ -1,3 +1,4 @@
|
||||
import json
|
||||
import pytest
|
||||
import shlex
|
||||
|
||||
@ -63,9 +64,11 @@ def test_parse() -> None:
|
||||
"""
|
||||
must parse string correctly
|
||||
"""
|
||||
assert PkgbuildPatch.parse("VALUE") == "VALUE"
|
||||
assert PkgbuildPatch.parse("(ARRAY VALUE)") == ["ARRAY", "VALUE"]
|
||||
assert PkgbuildPatch.parse("""("QU'OUTED" ARRAY VALUE)""") == ["QU'OUTED", "ARRAY", "VALUE"]
|
||||
assert PkgbuildPatch.parse("key", "VALUE").value == "VALUE"
|
||||
assert PkgbuildPatch.parse("key", "(ARRAY VALUE)").value == ["ARRAY", "VALUE"]
|
||||
assert PkgbuildPatch.parse("key", """("QU'OUTED" ARRAY VALUE)""").value == ["QU'OUTED", "ARRAY", "VALUE"]
|
||||
assert PkgbuildPatch.parse("key()", """{ function with " quotes }""").value == """{ function with " quotes }"""
|
||||
assert PkgbuildPatch.parse("key", json.dumps(["array", "value"])).value == ["array", "value"]
|
||||
|
||||
|
||||
def test_unquote() -> None:
|
||||
|
Loading…
Reference in New Issue
Block a user