Skip to content

Commit 6bb9189

Browse files
author
Ubuntu
committed
simplify wrapper
1 parent ac45e07 commit 6bb9189

1 file changed

Lines changed: 3 additions & 5 deletions

File tree

config/main.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1893,8 +1893,7 @@ def portchannel(db, ctx, namespace):
18931893
def add_portchannel(ctx, portchannel_name, min_links, fallback):
18941894
"""Add port channel"""
18951895

1896-
db = ctx.obj['db']
1897-
validated_db = ValidatedConfigDBConnector(db)
1896+
db = ValidatedConfigDBConnector(ctx.obj['db'])
18981897

18991898
if is_portchannel_present_in_db(db, portchannel_name):
19001899
ctx.fail("{} already exists!".format(portchannel_name))
@@ -1906,16 +1905,15 @@ def add_portchannel(ctx, portchannel_name, min_links, fallback):
19061905
fvs['min_links'] = str(min_links)
19071906
if fallback != 'false':
19081907
fvs['fallback'] = 'true'
1909-
validated_db.set_entry('PORTCHANNEL', portchannel_name, fvs)
1908+
db.set_entry('PORTCHANNEL', portchannel_name, fvs)
19101909

19111910
@portchannel.command('del')
19121911
@click.argument('portchannel_name', metavar='<portchannel_name>', required=True)
19131912
@click.pass_context
19141913
def remove_portchannel(ctx, portchannel_name):
19151914
"""Remove port channel"""
19161915

1917-
db = ctx.obj['db']
1918-
validated_db = ValidatedConfigDBConnector(db)
1916+
db = ValidatedConfigDBConnector(ctx.obj['db'])
19191917

19201918
if len([(k, v) for k, v in db.get_table('PORTCHANNEL_MEMBER') if k == portchannel_name]) != 0:
19211919
click.echo("Error: Portchannel {} contains members. Remove members before deleting Portchannel!".format(portchannel_name))

0 commit comments

Comments
 (0)