can now provision an IP address/subnet
This commit is contained in:
parent
5b79ea5157
commit
4ef0eb329c
93
automonty
93
automonty
@ -2,12 +2,12 @@
|
||||
|
||||
import argparse
|
||||
import librouteros
|
||||
import progress.bar
|
||||
import ssl
|
||||
import os
|
||||
import yaml
|
||||
import re
|
||||
import dns.resolver, dns.reversename
|
||||
import ipaddress
|
||||
|
||||
# comment format:
|
||||
# hostname.example.com [aM flag1 flag2 key1=value1 key2=option3:value3,option4:value4] selector.example.com
|
||||
@ -94,15 +94,13 @@ def router_in_groups( options, groups ):
|
||||
def connect_routers( config, args ):
|
||||
routers = config.get( 'router', {} )
|
||||
rval = {}
|
||||
with progress.bar.PixelBar( 'Connecting', max = len( routers ) ) as bar:
|
||||
for ( name, options ) in routers.items():
|
||||
if ( not args.only_router and not args.only_group ) or ( name in args.only_router ) or router_in_groups( options, args.only_group ):
|
||||
kwargs = {}
|
||||
kwargs.update( options.get( 'connection', {} ) )
|
||||
if 'host' not in kwargs:
|
||||
kwargs[ 'host' ] = name
|
||||
rval[ name ] = connection( **kwargs )
|
||||
bar.next()
|
||||
for ( name, options ) in routers.items():
|
||||
if ( not args.only_router and not args.only_group ) or ( name in args.only_router ) or router_in_groups( options, args.only_group ):
|
||||
kwargs = {}
|
||||
kwargs.update( options.get( 'connection', {} ) )
|
||||
if 'host' not in kwargs:
|
||||
kwargs[ 'host' ] = name
|
||||
rval[ name ] = connection( **kwargs )
|
||||
return rval
|
||||
|
||||
def monty_check( config, args, routers ):
|
||||
@ -179,6 +177,69 @@ def monty_fixup( config, args, routers ):
|
||||
print( name, ":", item[ 'interface' ], new_comment )
|
||||
ipv6_address.update( **{ 'comment': new_comment, '.id': item[ '.id' ] } )
|
||||
|
||||
def monty_provision( config, args, routers ):
|
||||
reverse = {}
|
||||
statics = {}
|
||||
print()
|
||||
for ( name, api ) in routers.items():
|
||||
if args.dry_run:
|
||||
print( '#' * ( len( name ) + 2 ) )
|
||||
print( '#', name )
|
||||
print()
|
||||
vlans = api.path( 'interface', 'vlan' )
|
||||
ip_address = api.path( 'ip', 'address' )
|
||||
ipv6_address = api.path( 'ipv6', 'address' )
|
||||
for vlan in vlans:
|
||||
if vlan[ 'vlan-id' ] == args.vlan:
|
||||
rval = {}
|
||||
if args.static:
|
||||
rval[ 'static' ] = True
|
||||
disabled = 'no'
|
||||
elif name in args.home:
|
||||
rval[ 'home' ] = True
|
||||
disabled = 'no'
|
||||
else:
|
||||
disabled = 'yes'
|
||||
comment = make_comment( args.selector, rval, args.hostname )
|
||||
if args.dry_run:
|
||||
if args.ipv4.network.prefixlen == 32:
|
||||
print( '/ip address add interface="%s" address="%s" network="%s" disabled=%s comment="%s"' % ( vlan[ 'name' ], config[ 'default_ipv4_loopback' ], args.ipv4.network.network_address, disabled, comment ) )
|
||||
else:
|
||||
print( '/ip address add interface="%s" address="%s" disabled=%s comment="%s"' % ( vlan[ 'name' ], args.ipv4, disabled, comment ) )
|
||||
if args.ipv6:
|
||||
if args.static:
|
||||
print( '/ipv6 address add interface="%s" address="%s" disabled=%s comment="%s" advertise=no' % ( vlan[ 'name' ], args.ipv6, disabled, comment ) )
|
||||
else:
|
||||
print( '/ipv6 address add interface="%s" address="%s" disabled=%s comment="%s" eui-64=yes advertise=yes' % ( vlan[ 'name' ], args.ipv6.network, disabled, comment ) )
|
||||
|
||||
else:
|
||||
v4addr = { 'interface': vlan[ 'name' ],
|
||||
'disabled': disabled,
|
||||
'comment': comment,
|
||||
}
|
||||
if args.ipv4.network.prefixlen == 32:
|
||||
v4addr[ 'address' ] = config[ 'default_ipv4_loopback' ]
|
||||
v4addr[ 'network' ] = str( args.ipv4.network.network_address )
|
||||
else:
|
||||
v4addr[ 'address' ] = str( args.ipv4 )
|
||||
ip_address.add( **v4addr )
|
||||
|
||||
if args.ipv6:
|
||||
v6addr = { 'interface': vlan[ 'name' ],
|
||||
'disabled': disabled,
|
||||
'comment': comment,
|
||||
}
|
||||
if args.static:
|
||||
v6addr[ 'address' ] = str( args.ipv6 )
|
||||
else:
|
||||
v6addr[ 'address' ] = str( args.ipv6.network )
|
||||
v6addr[ 'eui-64' ] = 'yes'
|
||||
v6addr[ 'advertise' ] = 'yes'
|
||||
ipv6_address.add( **v6addr )
|
||||
|
||||
if args.dry_run:
|
||||
print()
|
||||
|
||||
def load_configuration():
|
||||
try:
|
||||
config = open( os.path.expanduser( '~/.automonty.yaml' ), 'r' )
|
||||
@ -196,7 +257,7 @@ def main():
|
||||
|
||||
parser.add_argument( '--only-router', action = 'append', default = [] )
|
||||
parser.add_argument( '--only-group', action = 'append', default = [] )
|
||||
parser.add_argument( '-n', '--dry-run', action = 'store_true' )
|
||||
parser.add_argument( '-n', '--dry-run', action = 'store_true', default = False )
|
||||
|
||||
parser_check = subparsers.add_parser( 'check' )
|
||||
parser_check.add_argument( 'addr', nargs = "*" )
|
||||
@ -205,6 +266,16 @@ def main():
|
||||
parser_fixup = subparsers.add_parser( 'fixup' )
|
||||
parser_fixup.set_defaults( func = monty_fixup )
|
||||
|
||||
parser_provision = subparsers.add_parser( 'provision' )
|
||||
parser_provision.add_argument( '--home', action = 'append', default = [] )
|
||||
parser_provision.add_argument( '-s', '--static', action = 'store_true', default = False )
|
||||
parser_provision.add_argument( '--selector', type = str, default = None )
|
||||
parser_provision.add_argument( 'hostname' )
|
||||
parser_provision.add_argument( 'vlan', type = int )
|
||||
parser_provision.add_argument( 'ipv4', type = ipaddress.IPv4Interface )
|
||||
parser_provision.add_argument( 'ipv6', type = ipaddress.IPv6Interface, nargs = "?" )
|
||||
parser_provision.set_defaults( func = monty_provision )
|
||||
|
||||
args = parser.parse_args()
|
||||
routers = connect_routers( config, args )
|
||||
args.func( config, args, routers )
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user