@@ -36,23 +36,23 @@ def get_storage_key():
36
36
return storage_keys ['key1' ]
37
37
38
38
39
- def print_tokens (sa_key ):
39
+ def print_ip (sa_key ):
40
40
global PARTITION_NAME , ROW_ID , SA_NAME , TBL_NAME
41
41
tbl_svc = TableService (account_name = SA_NAME , account_key = sa_key )
42
42
if not tbl_svc .exists (TBL_NAME ):
43
43
return False
44
44
try :
45
45
token = tbl_svc .get_entity (TBL_NAME , PARTITION_NAME , ROW_ID )
46
- print '{}|{}|{} ' .format (token .manager_ip , token . manager_token , token . worker_token )
46
+ print '{}' .format (token .manager_ip )
47
47
return True
48
48
except :
49
49
return False
50
50
51
51
52
- def insert_tokens (sa_key , manager_ip , manager_token , worker_token ):
52
+ def insert_ip (sa_key , manager_ip ):
53
53
global PARTITION_NAME , ROW_ID , TBL_NAME , SA_NAME
54
54
tbl_svc = TableService (account_name = SA_NAME , account_key = sa_key )
55
- token = {'PartitionKey' : PARTITION_NAME , 'RowKey' : ROW_ID , 'manager_ip' : manager_ip , 'manager_token' : manager_token , 'worker_token' : worker_token }
55
+ token = {'PartitionKey' : PARTITION_NAME , 'RowKey' : ROW_ID , 'manager_ip' : manager_ip }
56
56
try :
57
57
# this upsert operation should always succeed
58
58
tbl_svc .insert_or_replace_entity (TBL_NAME , token )
@@ -80,11 +80,9 @@ def main():
80
80
parser = argparse .ArgumentParser (description = 'Tool to store Docker Swarm info in Azure Tables' )
81
81
subparsers = parser .add_subparsers (help = 'commands' , dest = 'action' )
82
82
create_table_parser = subparsers .add_parser ('create-table' , help = 'Create table specified in env var AZURE_TBL_NAME' )
83
- get_tokens_parser = subparsers .add_parser ('get-tokens ' , help = 'Get swarm info from table specified in env var AZURE_TBL_NAME' )
84
- insert_tokens_parser = subparsers .add_parser ('insert-tokens ' , help = 'Insert swarm info to table specified in env var AZURE_TBL_NAME' )
83
+ get_tokens_parser = subparsers .add_parser ('get-ip ' , help = 'Get swarm info from table specified in env var AZURE_TBL_NAME' )
84
+ insert_tokens_parser = subparsers .add_parser ('insert-ip ' , help = 'Insert swarm info to table specified in env var AZURE_TBL_NAME' )
85
85
insert_tokens_parser .add_argument ('ip' , help = 'IP address of the primary swarm manager' )
86
- insert_tokens_parser .add_argument ('manager_token' , help = 'Manager Token for the swarm' )
87
- insert_tokens_parser .add_argument ('worker_token' , help = 'Worker token for the swarm' )
88
86
89
87
args = parser .parse_args ()
90
88
@@ -93,10 +91,10 @@ def main():
93
91
if args .action == 'create-table' :
94
92
if not create_table (key ):
95
93
sys .exit (1 )
96
- elif args .action == 'get-tokens ' :
97
- print_tokens (key )
98
- elif args .action == 'insert-tokens ' :
99
- insert_tokens (key , args .ip , args . manager_token , args . worker_token )
94
+ elif args .action == 'get-ip ' :
95
+ print_ip (key )
96
+ elif args .action == 'insert-ip ' :
97
+ insert_ip (key , args .ip )
100
98
else :
101
99
parser .print_usage ()
102
100
0 commit comments