Python wrapper.Wrapper类代码示例

    def checkArgc():
        """ Check argument count """

        if len(sys.argv) == 1:
            Wrapper.error('-H for help and usage')

        return

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

    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

    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

 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__)

    def __init__(self, url):
        """ init """

        self.url = url
        self.token = ''

        Wrapper.disableWarnings()

        return

    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

 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()

 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()

    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

 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__)

 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)

    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

    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

    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