本文整理汇总了Python中wrapper.Wrapper类的典型用法代码示例。如果您正苦于以下问题:Python Wrapper类的具体用法?Python Wrapper怎么用?Python Wrapper使用的例子?那么恭喜您, 这里精选的类代码示例或许可以为您提供帮助。
在下文中一共展示了Wrapper类的15个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: checkArgc
def checkArgc():
""" Check argument count """
if len(sys.argv) == 1:
Wrapper.error('-H for help and usage')
return
示例2: Waiter
class Waiter(object):
"""waiter"""
def __init__(self):
self.wrap = Wrapper()
def add_order(self, waiter_id, order_data):
"""must put waiter id and tickets in list of dict
[{"id_dish": 1, "count": 1},{"id_dish": 12, "count": 1}..."""
date = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
order = {"id_status": 4,
"id_user": waiter_id,
"date": date
}
order_id = self.wrap.insert(order, "orders")
for ticket in order_data:
ticket["id_order"] = order_id
try:
del ticket["image"]
del ticket["name"]
del ticket["id"]
del ticket["description"]
del ticket["id_category"]
except Exception, e:
print e
self.wrap.insert(ticket, "tickets")
return order_id
示例3: start
def start(self):
""" All program flow is handled here """
# usage, init, checks
Usage.banner()
args = ArgsParser.parseArgs()
self.opts = vars(args)
Check.checkArgc()
Check.checkArgs(['-n', '-m'])
Check.checkActions(self.opts['action'])
Check.checkReportFormat(self.opts['format'])
# ini config
Wrapper.msg("Reading configuration file " + "'" +
self.opts['config'] + "'")
self.ini = IniConfig(self.opts['config'])
self.ini.readConfig()
Check.checkIniFile(self.ini.config)
# additional options from config file
self.getOptions()
### nessus actions ###
# nessus report
if self.opts['action'] in ACTIONS:
self.processReports(self.opts['action'])
# send reports
if self.opts['mail']:
if self.opts['action'] in ACTIONS:
self.sendReports(self.opts['action'])
return
示例4: sendRequest
def sendRequest(self, method, resource, data='', token='', verify=False):
""" Send a request to nessus based on the specified data """
resource = self.url + resource
headers = { 'X-Cookie': 'token={0}'.format(self.token),
'Content-Type': 'application/json' }
data = json.dumps(data)
if method == 'POST':
r = requests.post(resource, data=data, headers=headers,
verify=verify)
elif method == 'PUT':
r = requests.put(resource, data=data, headers=headers,
verify=verify)
elif method == 'DELETE':
r = requests.delete(resource, data=data, headers=headers,
verify=verify)
else:
r = requests.get(resource, params=data, headers=headers,
verify=verify)
if r.status_code != 200:
e = r.json()
Wrapper.error(e['error'])
if 'download' in resource:
return r.content
# temp dirty fix. TODO: make it better
elif 'session' in resource and method == 'DELETE':
return
else:
return r.json()
return
示例5: __init__
def __init__(self, *handlers, **kwargs):
super(Container, self).__init__(*handlers, **kwargs)
if "impl" not in kwargs:
self.ssl = SSLConfig()
self.global_handler = GlobalOverrides(kwargs.get("global_handler", self.global_handler))
self.trigger = None
self.container_id = str(generate_uuid())
Wrapper.__setattr__(self, "subclass", self.__class__)
示例6: __init__
def __init__(self, url):
""" init """
self.url = url
self.token = ''
Wrapper.disableWarnings()
return
示例7: checkReportFormat
def checkReportFormat(rformat):
""" Check if chosen report format exists """
formats = ('html', 'pdf')
if rformat not in formats:
Wrapper.error('Wrong report format: %s' % rformat)
return
示例8: __init__
def __init__(self, parent=None):
super(ScudCloud, self).__init__(parent)
self.setWindowTitle('ScudCloud')
self.notifier = Notifier(self.APP_NAME, get_resource_path('scudcloud.png'))
self.settings = QSettings(expanduser("~")+"/.scudcloud", QSettings.IniFormat)
self.identifier = self.settings.value("Domain")
if Unity is not None:
self.launcher = Unity.LauncherEntry.get_for_desktop_id("scudcloud.desktop")
else:
self.launcher = DummyLauncher(self)
self.leftPane = LeftPane(self)
self.cookiesjar = PersistentCookieJar(self)
webView = Wrapper(self)
webView.page().networkAccessManager().setCookieJar(self.cookiesjar)
self.stackedWidget = QtGui.QStackedWidget()
self.stackedWidget.addWidget(webView)
centralWidget = QtGui.QWidget(self)
layout = QtGui.QHBoxLayout()
layout.setContentsMargins(0, 0, 0, 0)
layout.setSpacing(0)
layout.addWidget(self.leftPane)
layout.addWidget(self.stackedWidget)
centralWidget.setLayout(layout)
self.setCentralWidget(centralWidget)
self.addMenu()
self.tray = Systray(self)
self.systray()
self.installEventFilter(self)
self.zoom()
if self.identifier is None:
webView.load(QtCore.QUrl(self.SIGNIN_URL))
else:
webView.load(QtCore.QUrl(self.domain()))
webView.show()
示例9: __init__
def __init__(self, parent = None, settings_path = ""):
super(zcswebapp, self).__init__(parent)
self.setWindowTitle('zcswebapp')
self.settings_path = settings_path
self.notifier = Notifier(Resources.APP_NAME, Resources.get_path('zcswebapp.png'))
self.settings = QSettings(self.settings_path + '/zcswebapp.cfg', QSettings.IniFormat)
self.identifier = self.settings.value("Domain")
if Unity is not None:
self.launcher = Unity.LauncherEntry.get_for_desktop_id("zcswebapp.desktop")
else:
self.launcher = DummyLauncher(self)
self.webSettings()
self.leftPane = LeftPane(self)
webView = Wrapper(self)
webView.page().networkAccessManager().setCookieJar(self.cookiesjar)
self.stackedWidget = QtGui.QStackedWidget()
self.stackedWidget.addWidget(webView)
centralWidget = QtGui.QWidget(self)
layout = QtGui.QHBoxLayout()
layout.setContentsMargins(0, 0, 0, 0)
layout.setSpacing(0)
layout.addWidget(self.leftPane)
layout.addWidget(self.stackedWidget)
centralWidget.setLayout(layout)
self.setCentralWidget(centralWidget)
self.addMenu()
self.tray = Systray(self)
self.systray(zcswebapp.minimized)
self.installEventFilter(self)
if self.identifier is None:
webView.load(QtCore.QUrl(Resources.SIGNIN_URL))
else:
webView.load(QtCore.QUrl(self.domain()))
webView.show()
示例10: sendReports
def sendReports(self, option):
""" Send out reports via email """
msg = {} # mail header and body
mail = ''
server = {} # smtp server
m = Mail()
# smtp server address and tcp port
server['addr'] = self.opts['smtp_host']
server['port'] = self.opts['smtp_port']
# smtp (static content) read from ini file
msg['from'] = self.opts['smtp_from']
Wrapper.msg('Sending out reports')
if option == 'report':
# for each report in report directory except summary.txt
for f in os.listdir(self.opts['outdir'] + '/' + TODAY):
if f != 'summary.txt':
report = self.opts['outdir'] + '/' + TODAY + '/' + f
name = f.split('.')[0]
# mail header + body
msg['message'] = 'Hi,\n\nplease find attached the Nessus ' \
'report for this week.\n\nBest Regards,\n\n' \
"Rocket Internet's Security Team"
msg['subject'] = '[{0}] Your new Nessus report for {1} ' \
'is ready'.format(name, TODAY)
msg['to'] = self.ini.config.get('addressbook', name)
# build and send mail
mail = m.buildMail(msg, (report,))
m.sendMail(server, mail, self.opts['smtp_login'], tls=True)
else:
report = self.opts['outdir'] + '/' + TODAY + '/' + 'summary.txt'
with open(report, 'r') as f:
report_data = f.read()
# mail header + body
msg['message'] = 'Hi,\n\nplease find below the Nessus ' \
'Summary Report for this week:\n\n'
msg['message'] += report_data
msg['message'] += "\n\nBest Regards,\n\nRocket Internet's " \
"Security Team"
msg['subject'] = 'Nessus Summary Report ({0})'.format(TODAY)
msg['to'] = self.ini.config.get('addressbook', 'Summary')
# build and send mail
mail = m.buildMail(msg, (report,))
m.sendMail(server, mail, self.opts['smtp_login'], tls=True)
return
示例11: __init__
def __init__(self, *handlers, **kwargs):
super(Container, self).__init__(*handlers, **kwargs)
if "impl" not in kwargs:
try:
self.ssl = SSLConfig()
except SSLUnavailable:
self.ssl = None
self.global_handler = GlobalOverrides(kwargs.get('global_handler', self.global_handler))
self.trigger = None
self.container_id = str(generate_uuid())
Wrapper.__setattr__(self, 'subclass', self.__class__)
示例12: addWrapper
def addWrapper(self, url):
webView = Wrapper(self)
webView.page().networkAccessManager().setCookieJar(self.cookiesjar)
webView.page().networkAccessManager().setCache(self.diskCache)
webView.load(QtCore.QUrl(url))
webView.show()
self.stackedWidget.addWidget(webView)
self.stackedWidget.setCurrentWidget(webView)
示例13: checkArgs
def checkArgs(args):
""" Check if required arguments are given """
length = len(args)
for a in args:
if a not in sys.argv:
length -= 1
if length is 0:
Wrapper.error('Wrong usage. -H for help.')
return
示例14: checkIniFile
def checkIniFile(ini):
""" Check ini file configuration """
sections = ini.sections()
if not sections:
Wrapper.error('Cannot read from ini file. Empty or wrong path?')
check = ('nessus', 'mail', 'addressbook')
for s in sections:
if s not in check:
Wrapper.warn('Missing or wrong section: [%s]' % s)
return
示例15: getNessusLogin
def getNessusLogin(self):
""" get Nessus login from cmdline or read from ini config file """
login = {}
if self.opts['nessus_login']:
login['user'] = self.opts['nessus_login'].split(':')[0]
login['pass'] = self.opts['nessus_login'].split(':')[1]
elif self.ini.config['nessus']:
login['user'] = self.ini.config.get('nessus', 'user')
login['pass'] = self.ini.config.get('nessus', 'pass')
else:
Wrapper.error('no login specified')
return login