예제 #1
0
    def test_load(self):
        self.maxDiff = None
        data = {
            "branch": "stable",
            "date": "2020-05-04 22:25:45+02:00",
            "priority": "1",
            "body":
            """8devices-carambola2-board 2020.1.2-242-testing c8a7d3db620e91ad4d935bfc3df59f1a60bb3a2f11aa19a24b2f3d965cb88d06 4259844 gluon-ffsh-2020.1.2-242-testing-8devices-carambola2-board-sysupgrade.bin
8devices-carambola2-board 2020.1.2-242-testing c8a7d3db620e91ad4d935bfc3df59f1a60bb3a2f11aa19a24b2f3d965cb88d06 gluon-ffsh-2020.1.2-242-testing-8devices-carambola2-board-sysupgrade.bin
8devices-carambola2-board 2020.1.2-242-testing cffca7ec6dee0f4e1cb95fe79275a34cf791b92f76ba74e4d3e74c113e7eed40e5ae576c7f8da1e9cdc4dec71f14f8457111870647aafc5f41b048749d34b671 gluon-ffsh-2020.1.2-242-testing-8devices-carambola2-board-sysupgrade.bin
alfa-network-ap121f 2020.1.2-242-testing a21015a1cc0c46201e09fcdb87a9b03d9ecd0169ff1f8cab97a7e8b39767b87e 4326145 gluon-ffsh-2020.1.2-242-testing-alfa-network-ap121f-sysupgrade.bin""",
            "signature": "4b93f300b95e89342b2d49dc5fe4fbf1d89028ad48e00a09ecc2af86beb6380fea1cd8aa9f9f05a740dfa18589e7223a138920b7eb0639914b473cf4f37e7b04"
        }
        manifest = Manifest(data["branch"], data["date"], data["priority"], data["body"], data["signature"])
        manifest.export("stable.manifest")
        with open("stable.manifest", "r") as manifest_file:
            self.assertTrue(manifest_file.readlines())
        manifest2 = Manifest()
        manifest2.load("stable.manifest")
        manifest2.export("stable2.manifest")
        self.assertEqual(manifest2.branch, data["branch"])
        self.assertEqual(manifest2.date, data["date"])
        self.assertEqual(manifest2.priority, data["priority"])
        self.assertEqual(manifest2.body, data["body"])
        self.assertEqual(manifest2.signature, data["signature"])
예제 #2
0
 def manifest(self, version, branch):
     version_pattern = re.compile(r'(\d{4}.\d.\d).(\d)')
     if version_pattern.match(version) and branch is not None:
         manifest = Manifest()
         manifest_path = Path(self.config["firmware_path"]
                              ) / version / "sysupgrade" / "stable.manifest"
         manifest.load(manifest_path)
         manifest.set_branch(branch)
         manifest.export(manifest_path.parent /
                         "{}.{}".format(branch, "manifest"))
예제 #3
0
    def save(self, artifact_file):
        filename = Path(artifact_file).name
        filename_parsed = re.match(r'(\d{4}.\d.\d).(\d)_(.*)+_(\w+).zip', filename)

        release_name = "{}.{}".format(filename_parsed.group(1), filename_parsed.group(2))
        release_dir = self.storage_path / (release_name)

        if not release_dir.exists():
            release_dir.mkdir()

        with zipfile.ZipFile(artifact_file, 'r') as zip_ref:
            for file in zip_ref.infolist():
                if file.filename.endswith("stable.manifest"):
                    file.filename = Path(file.filename).name

                    manifest_path = release_dir / "sysupgrade" / "stable.manifest"
                    if manifest_path.exists():
                        zip_ref.extract(file, "/tmp")
                        manifest = Manifest()
                        manifest.load(manifest_path)
                        manifest_part = Manifest()
                        manifest_part.load(Path("/tmp") / "stable.manifest")
                        manifest.merge(manifest_part)
                        manifest.export(manifest_path)
                    else:
                        zip_ref.extract(file, manifest_path.parent)

                if file.filename.startswith('images/factory/'):
                    file.filename = Path(file.filename).name
                    if file.filename != "factory":
                        zip_ref.extract(file, release_dir / "factory")
                elif file.filename.startswith('images/sysupgrade/'):
                    file.filename = Path(file.filename).name
                    if file.filename != "sysupgrade":
                        zip_ref.extract(file, release_dir / "sysupgrade")
                elif file.filename.startswith('images/other/'):
                    file.filename = Path(file.filename).name
                    if file.filename != "other":
                        zip_ref.extract(file, release_dir / "other")