-
Notifications
You must be signed in to change notification settings - Fork 0
/
ImportExport.py
127 lines (116 loc) · 4.14 KB
/
ImportExport.py
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
# Copyright (C) 2012-2020 MIPS Tech LLC
# Written by Matthew Fortune <[email protected]> and
# Daniel Sanders <[email protected]>
# This file is part of Overtest.
#
# Overtest is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3, or (at your option)
# any later version.
#
# Overtest is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with overtest; see the file COPYING. If not, write to the Free
# Software Foundation, 51 Franklin Street - Fifth Floor, Boston, MA
# 02110-1301, USA.
import sys
import os
import getopt
from OvertestExceptions import *
import importers
import exporters
import OvtDB
class ImportExport:
ovtDB = None
def __init__(self, _ovtDB):
self.ovtDB = _ovtDB
def usage(self, exitcode, error = None):
"""
Display the usage
"""
if error != None:
self.error(error)
print ""
print "Usage:"
print " --schema=<schema>"
print " <options>"
sys.exit (exitcode)
def exporter(self, args):
e = self.getSchemaClass (args, False)
return e.exportData (args[1:])
def importer(self, args):
i = self.getSchemaClass (args, True)
return i.importData (args[1:])
def getSchemaClass(self, args, importing):
"""
Parse the common options for import or export
"""
if len(args) == 0:
self.usage(2, "Schema not specified")
try:
opts, args = getopt.getopt (args[0:1], "s:", ["schema="])
except getopt.GetoptError, e:
self.usage (2, str(e))
for (o, a) in opts:
if o in ("-s", "--schema"):
if a.lower() == "action":
if importing:
import importers.ActionImporter
return importers.ActionImporter.ActionImporter(self.ovtDB)
else:
import exporters.ActionExporter
return exporters.ActionExporter.ActionExporter(self.ovtDB)
elif a.lower() == "hostconfig":
if importing:
self.error ("The hostconfig schema cannot be imported")
else:
import exporters.HostConfigExporter
return exporters.HostConfigExporter.HostConfigExporter(self.ovtDB)
elif a.lower() == "results":
if importing:
self.error ("The results schema cannot be imported")
else:
import exporters.ResultsExporter
return exporters.ResultsExporter.ResultsExporter(self.ovtDB)
elif a.lower() == "log":
if importing:
self.error ("The log schema cannot be imported")
else:
import exporters.LogExporter
return exporters.LogExporter.LogExporter(self.ovtDB)
elif a.lower() == "notify":
if importing:
self.error ("The notify schema cannot be imported")
else:
import exporters.NotifyExporter
return exporters.NotifyExporter.NotifyExporter(self.ovtDB)
elif a.lower() == "testrunid":
if importing:
self.error ("The testrunid schema cannot be imported")
else:
import exporters.TestrunIdExporter
return exporters.TestrunIdExporter.TestrunIdExporter(self.ovtDB)
elif a.lower() == "testrun":
if importing:
self.error ("The testrun schema cannot be imported")
else:
import exporters.TestrunExporter
return exporters.TestrunExporter.TestrunExporter(self.ovtDB)
elif a.lower() == "claim":
if importing:
self.error ("The claim schema cannot be imported")
else:
import exporters.ClaimExporter
return exporters.ClaimExporter.ClaimExporter(self.ovtDB)
else:
self.usage(1, "Unknown schema")
self.usage(2, "Schema not specified")
def error(self, error):
"""
Print an error message
"""
print "ERROR: %s"%error