blob: 23b91e1bc1902138bcbddf62fad2504789cc6f4d (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
import Bcfg2.Server.Admin
import sqlalchemy, sqlalchemy.orm
import Bcfg2.Server.Snapshots
import Bcfg2.Server.Snapshots.model
class Snapshots(Bcfg2.Server.Admin.Mode):
__shorthelp__ = "Interact with the Snapshots system"
__longhelp__ = (__shorthelp__)
__usage__ = ("bcfg2-admin snapshots [init|query qtype] ")
q_dispatch = {'client':Bcfg2.Server.Snapshots.model.Client,
'group':Bcfg2.Server.Snapshots.model.Group,
'snapshot':Bcfg2.Server.Snapshots.model.Snapshot,
}
def __init__(self, configfile):
Bcfg2.Server.Admin.Mode.__init__(self, configfile)
self.session = Bcfg2.Server.Snapshots.setup_session(debug=True)
def __call__(self, args):
if args[0] == 'query':
if args[1] in self.q_dispatch:
q_obj = self.q_dispatch[args[1]]
results = self.session.query(q_obj).all()
else:
print 'error'
raise SystemExit, 1
for result in results:
print result.name
elif args[0] == 'init':
dbpath = Bcfg2.Server.Snapshots.db_from_config()
engine = sqlalchemy.create_engine(dbpath, echo=True)
metadata = Base.metadata
metadata.create_all(engine)
Session = sessionmaker()
Session.configure(bind=engine)
session = Session()
session.commit()
|