Current Dir: /snap/core22/2437/usr/share/subiquity/subiquitycore/
[DIR] controllers [ delete | rename ]
[DIR] models [ delete | rename ]
[DIR] testing [ delete | rename ]
[DIR] tests [ delete | rename ]
[DIR] ui [ delete | rename ]
[DIR] __pycache__ [ delete | rename ]
[FILE] async_helpers.py [ edit | delete | rename ]
[FILE] context.py [ edit | delete | rename ]
[FILE] controller.py [ edit | delete | rename ]
[FILE] controllerset.py [ edit | delete | rename ]
[FILE] core.py [ edit | delete | rename ]
[FILE] file_util.py [ edit | delete | rename ]
[FILE] i18n.py [ edit | delete | rename ]
[FILE] log.py [ edit | delete | rename ]
[FILE] lsb_release.py [ edit | delete | rename ]
[FILE] netplan.py [ edit | delete | rename ]
[FILE] palette.py [ edit | delete | rename ]
[FILE] prober.py [ edit | delete | rename ]
[FILE] pubsub.py [ edit | delete | rename ]
[FILE] screen.py [ edit | delete | rename ]
[FILE] snapd.py [ edit | delete | rename ]
[FILE] ssh.py [ edit | delete | rename ]
[FILE] tui.py [ edit | delete | rename ]
[FILE] tuicontroller.py [ edit | delete | rename ]
[FILE] utils.py [ edit | delete | rename ]
[FILE] view.py [ edit | delete | rename ]
[FILE] __init__.py [ edit | delete | rename ]
Viewing: /snap/core22/2437/usr/share/subiquity/subiquitycore/prober.py
# Copyright 2015 Canonical, Ltd.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, version 3.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# 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 logging
import time
import yaml
from probert.network import (
StoredDataObserver,
UdevObserver,
)
log = logging.getLogger('subiquitycore.prober')
class Prober():
def __init__(self, machine_config, debug_flags):
self.saved_config = None
if machine_config:
self.saved_config = yaml.safe_load(machine_config)
self.debug_flags = debug_flags
log.debug('Prober() init finished, data:{}'.format(self.saved_config))
def probe_network(self, receiver):
if self.saved_config is not None:
observer = StoredDataObserver(
self.saved_config['network'], receiver)
else:
observer = UdevObserver(receiver)
return observer, observer.start()
def get_storage(self, probe_types=None):
if self.saved_config is not None:
flag = 'bpfail-full'
if probe_types is not None:
flag = 'bpfail-restricted'
if flag in self.debug_flags:
time.sleep(2)
1/0
r = self.saved_config['storage'].copy()
if probe_types is not None:
for k in self.saved_config['storage']:
if k not in probe_types:
r[k] = {}
return r
from probert.storage import Storage
return Storage().probe(probe_types=probe_types)