1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
# Copyright (c) 2020 Egor Tensin <Egor.Tensin@gmail.com>
# This file is part of the "cmake-common" project.
# For details, see https://github.com/egor-tensin/cmake-common.
# Distributed under the MIT License.
import abc
from contextlib import contextmanager
import logging
import os.path
import shutil
from project.boost.directory import BoostDir
from project.utils import temp_file
class Archive:
def __init__(self, version, path):
self.version = version
self.path = path
@property
def dir_name(self):
return self.version.dir_name
def unpack(self, dest_dir):
path = os.path.join(dest_dir, self.dir_name)
if os.path.exists(path):
raise RuntimeError(f'Boost directory already exists: {path}')
logging.info('Unpacking Boost to: %s', path)
shutil.unpack_archive(self.path, dest_dir)
return BoostDir(path)
class ArchiveStorage(abc.ABC):
@abc.abstractmethod
def get_archive(self, version):
pass
@contextmanager
@abc.abstractmethod
def write_archive(self, version, contents):
pass
class PermanentStorage(ArchiveStorage):
def __init__(self, cache_dir):
self._dir = cache_dir
def _archive_path(self, version):
return os.path.join(self._dir, version.archive_name)
def get_archive(self, version):
path = self._archive_path(version)
if os.path.exists(path):
return path
return None
@contextmanager
def write_archive(self, version, contents):
path = self._archive_path(version)
logging.info('Writing Boost archive: %s', path)
if os.path.exists(path):
raise RuntimeError(f'cannot download Boost, file already exists: {path}')
with open(path, mode='w+b') as dest:
dest.write(contents)
yield path
class TemporaryStorage(ArchiveStorage):
def __init__(self, temp_dir):
self._dir = temp_dir
def get_archive(self, version):
return None
@contextmanager
def write_archive(self, version, contents):
tmp = temp_file(prefix=f'boost_{version}_',
suffix=version.archive_ext, dir=self._dir)
with tmp as archive_path:
with open(archive_path, mode='wb') as fd:
fd.write(contents)
yield archive_path
|
