facebookincubator / fcr Goto Github PK

future: <Task finished coro=<SSHConnection._run_task() done, defined at /Users/bthornto/github/FCR2/venv/lib/python3.6/site-packages/asyncssh/connection.py:366> exception=CancelledError()>
concurrent.futures._base.CancelledError

            if self.check_ip(ip):
                return ip.addr

            if self.check_ip(ip):
                return ip

    message='run failed: AttributeError("\'str\' object has no attribute \'addr\'",)')

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

# Copyright (c) 2017-present, Facebook, Inc.
# All rights reserved.
#
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree. An additional grant
# of patent rights can be found in the PATENTS file in the same directory.
#

from fbnet.command_runner.service import FcrServiceBase
from fbnet.command_runner.command_server import CommandServer
from fbnet.command_runner.device_db import BaseDeviceDB
from fbnet.command_runner.device_vendor import DeviceVendors
from fbnet.command_runner.device_info import DeviceInfo, DeviceIP

from fbnet.command_runner.device_db import BaseDeviceDB
from fbnet.command_runner.options import Option

# You will likely get this data from your backend systems.
# But for illustration we will assume this is available in JSON
# format
import json
json_devdata = '''
[
  {"host": "device.company.net", "chassis": "N7000", "ip": ["10.250.255.15"], "role": "switch", "vendor": "cisco_nxos"}
]
'''

class DeviceDB(BaseDeviceDB):

    async def _fetch_device_data(self, name_filter=None, hostname=None):
        '''
        Fetch data from your backend database.

        This sample implementation assumes you have the data in a json format
        '''
        devinfos = json.loads(json_devdata)

        return {self._make_dev(devinfo) for devinfo in devinfos }

    def _make_dev(self, devinfo):
        return DeviceInfo(
            self.service,
            hostname=devinfo['host'],
            username='default',     # typically a user with a bare minimum RO priviledges
            password='passwd',
            pref_ips=devinfo['ip'], # a list of IP addresses in order of preferences
            ip=devinfo['ip'][0],    # the default fallback IP (after exhausting the pref_ips
            vendor_data=self.service.vendors.get(devinfo['vendor']),
            # vendor_data=self.app.vendors.get(devinfo['vendor']),
            role=devinfo['role'],
            ch_model=devinfo['chassis'])

class Vendors(DeviceVendors):
    pass


class FCRService(FcrServiceBase):

    def __init__(self):

        super().__init__("FCR")

        self.vendors = Vendors(self)
        self.device_db = DeviceDB(self)
        self.service = CommandServer(self)


def main():

    service = FCRService()
    service.start()


if __name__ == "__main__":
    main()