OLD | NEW |
(Empty) | |
| 1 # Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. |
| 4 |
| 5 import os |
| 6 |
| 7 from catapult_base.dependency_manager import exceptions |
| 8 from catapult_base.dependency_manager import dependency_manager_util |
| 9 |
| 10 |
| 11 class ArchiveInfo(object): |
| 12 |
| 13 def __init__(self, archive_file, unzip_path, path_within_archive): |
| 14 """ Container for the information needed to unzip a downloaded archive. |
| 15 |
| 16 Args: |
| 17 archive_path: Path to the archive file. |
| 18 unzip_path: Path to unzip the archive into. Assumes that this path |
| 19 is unique for the archive. |
| 20 path_within_archive: Specify if and how to handle zip archives |
| 21 downloaded from cloud_storage. Expected values: |
| 22 None: Do not unzip the file downloaded from cloud_storage. |
| 23 '.': Unzip the file downloaded from cloud_storage. The |
| 24 unzipped file/folder is the expected dependency. |
| 25 file_path: Unzip the file downloaded from cloud_storage. |
| 26 |file_path| is the path to the expected dependency, |
| 27 relative to the unzipped archive path. |
| 28 """ |
| 29 self._archive_file = archive_file |
| 30 self._unzip_path = unzip_path |
| 31 self._path_within_archive = path_within_archive |
| 32 self._dependency_path = os.path.join( |
| 33 self._unzip_path, self._path_within_archive) |
| 34 if not self._has_minimum_data: |
| 35 raise ValueError( |
| 36 'Not enough information specified to initialize an archive info.' |
| 37 ' %s' % self) |
| 38 |
| 39 def GetUnzippedPath(self): |
| 40 if self.ShouldUnzipArchive(): |
| 41 # TODO(aiolos): Replace UnzipFile with zipfile.extractall once python |
| 42 # version 2.7.4 or later can safely be assumed. |
| 43 dependency_manager_util.UnzipArchive( |
| 44 self._archive_file, self._unzip_path) |
| 45 if self.ShouldUnzipArchive(): |
| 46 raise exceptions.ArchiveError( |
| 47 "Expected path '%s' was not extracted from archive '%s'." % |
| 48 (self._dependency_path, self._archive_file)) |
| 49 return self._dependency_path |
| 50 |
| 51 def ShouldUnzipArchive(self): |
| 52 if not self._has_minimum_data: |
| 53 raise exceptions.ArchiveError( |
| 54 'Missing needed info to unzip archive. Known data: %s', |
| 55 self.data_string) |
| 56 return not os.path.exists(self._dependency_path) |
| 57 |
| 58 @property |
| 59 def _has_minimum_data(self): |
| 60 return all([self._archive_file, self._unzip_path, |
| 61 self._dependency_path]) |
| 62 |
| 63 def __repr__(self): |
| 64 return ( |
| 65 'ArchiveInfo(archive_file=%s, unzip_path=%s, path_within_archive=%s, ' |
| 66 'dependency_path =%s)' % ( |
| 67 self._archive_file, self._unzip_path, self._path_within_archive, |
| 68 self._dependency_path)) |
| 69 |
OLD | NEW |