2 # vim: tabstop=4 shiftwidth=4 softtabstop=4
3 # -*- coding: utf-8 -*-
5 # ti2.py : List TAP device info for running VMs
8 # basecode "tapinfo.py" : List TAP device info for running VMs
15 from optparse import OptionParser, OptionGroup
16 from linkpair.collect.collector import Collector
17 from linkpair.linkpair import LinkPair
18 from linkpair.formatter.grapheasy import GraphEasyFormatter
19 from linkpair.formatter.springy import SpringyFormatter
20 from linkpair.formatter.springy_raphael import SpringyRaphaelFormatter
21 from linkpair.formatter.graphviz_tb import GraphvizTBFormatter
22 from linkpair.dbutils import DBUtils
25 def print_linkpair_records(lps):
26 format.set_linkpairs(lps)
27 for lp in format.format_linkpairs():
30 if __name__ == "__main__":
31 # init program options
32 p = OptionParser(version="ver:%s" % __version__)
33 g1 = OptionGroup(p, "Remote connection",
34 "connect to a remote system via SSH")
35 g1.add_option('-r', '--remote', dest='remote_desc',
36 help="specify user's name and hostname, port number (ex. -r user@hostname:22)", metavar="REMOTE_DESC")
37 g1.add_option('-p', '--password', dest='remote_password',
38 help="specify user's SSH password", metavar="PASSWORD")
39 g1.add_option('-k', '--sshkey', dest='remote_sshkey',
40 help="specify user's SSH private-key file", metavar="FILE")
41 p.add_option_group(g1)
42 p.add_option('-f', '--format', dest='format',
43 choices=['g', 'graph-easy', 's', 'springy',
44 'r', 'springy-raphael', 't', 'graphviz-tb'],
46 help="specify output format [FORMAT: g/graph-easy, s/springy, r/springy-raphael, t/graphviz-tb]",
48 p.add_option('-d', '--database', dest='database', default="",
49 help="store record to database", metavar="FILE")
50 p.add_option('-t', '--datatype', dest='datatype', default="json",
51 help="specify database type(default: json)", metavar="TYPE")
53 '-v', '--verbose', dest='verbose', action="store_true", default=False,
54 help="verbose output.")
55 (opts, args) = p.parse_args()
57 if opts.database is not None and opts.database != "":
59 # sql_conn = dbu.open_db(opts.database)
62 if opts.format == "g" or opts.format == "graph-easy":
63 format = GraphEasyFormatter()
64 elif opts.format == "s" or opts.format == "springy":
65 format = SpringyFormatter()
66 elif opts.format == "r" or opts.format == "springy-raphael":
67 format = SpringyRaphaelFormatter()
68 elif opts.format == "t" or opts.format == "graphviz-tb":
69 format = GraphvizTBFormatter()
71 format = GraphEasyFormatter()
73 col = Collector(opts.remote_desc, dbu, format)
74 if opts.remote_sshkey is not None:
75 col.set_remote_sshkey(opts.remote_sshkey)
76 if opts.remote_password is not None:
77 col.set_remote_password(opts.remote_password)
80 linkpairs = col.get_linkpairs()
82 dbu.store_to_db(opts.database, linkpairs)
84 print_linkpair_records(linkpairs)