2014-04-03 19:08:04 -03:00
|
|
|
from bootstrapvz.base import Task
|
|
|
|
|
from bootstrapvz.common import phases
|
|
|
|
|
from bootstrapvz.common.tasks import apt
|
2014-03-27 23:44:38 -03:00
|
|
|
import os
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class SetAptProxy(Task):
|
|
|
|
|
description = 'Setting proxy for APT'
|
|
|
|
|
phase = phases.package_installation
|
|
|
|
|
successors = [apt.AptUpdate]
|
|
|
|
|
|
|
|
|
|
@classmethod
|
|
|
|
|
def run(cls, info):
|
|
|
|
|
proxy_path = os.path.join(info.root, 'etc/apt/apt.conf.d/02proxy')
|
|
|
|
|
proxy_address = info.manifest.plugins['apt_proxy']['address']
|
|
|
|
|
proxy_port = info.manifest.plugins['apt_proxy']['port']
|
|
|
|
|
with open(proxy_path, 'w') as proxy_file:
|
|
|
|
|
proxy_file.write('Acquire::http {{ Proxy "http://{address}:{port}"; }};\n'
|
|
|
|
|
.format(address=proxy_address, port=proxy_port))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class RemoveAptProxy(Task):
|
|
|
|
|
description = 'Removing APT proxy configuration file'
|
|
|
|
|
phase = phases.system_cleaning
|
|
|
|
|
|
|
|
|
|
@classmethod
|
|
|
|
|
def run(cls, info):
|
|
|
|
|
os.remove(os.path.join(info.root, 'etc/apt/apt.conf.d/02proxy'))
|
