当前位置: 首页>>代码示例>>Python>>正文


Python Display.selectlist方法代码示例

本文整理汇总了Python中display.Display.selectlist方法的典型用法代码示例。如果您正苦于以下问题:Python Display.selectlist方法的具体用法?Python Display.selectlist怎么用?Python Display.selectlist使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在display.Display的用法示例。


在下文中一共展示了Display.selectlist方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。

示例1: Framework

# 需要导入模块: from display import Display [as 别名]
# 或者: from display.Display import selectlist [as 别名]

#.........这里部分代码省略.........
    #----------------------------
    # Select Web Templates
    #----------------------------
    def select_web_templates(self):
        templates = []

        # get lists of current templates
        db_static_templates = self.db.getWebTemplates(ttype="static")
        db_dynamic_templates = self.db.getWebTemplates(ttype="dynamic")
        
        # check to see if we have templates
        if (db_static_templates or db_dynamic_templates):
            for template in db_static_templates:
                parts = template.split("[-]")
                template_file = parts[0] + "/CONFIG"
                if Utils.is_readable(template_file) and os.path.isfile(template_file):
                    templates.append(("static", parts[0], parts[1]))
            for template in db_dynamic_templates:
                parts = template.split("[-]")
                template_file = parts[0] + "/CONFIG"
                if Utils.is_readable(template_file) and os.path.isfile(template_file):
                    templates.append(("dynamic", parts[0], parts[1]))
        else:
            # assume we do not have any valid templates
            # load all standard templates
            for f in os.listdir(self.config["web_template_path"]):
                template_file = os.path.join(self.config["web_template_path"], f) + "/CONFIG"
                if Utils.is_readable(template_file) and os.path.isfile(template_file):
                    templates.append(("static", os.path.join(self.config["web_template_path"], f), ""))
                    print "FIXED = [%s]" % (os.path.join(self.config["web_template_path"], f))

        # if "always yes" is enabled then just use all templates
        if (not self.config["always_yes"]):
            items = self.display.selectlist("Please select (comma seperated) the item(s) you wish to use. : ", templates)
            templates_temp = []
            self.db.clearWebTemplates()
            for item in items:
                print templates[int(item)-1]
                templates_temp.append(templates[int(item)-1])
                self.db.addWebTemplate(ttype=templates[int(item)-1][0], src_url=templates[int(item)-1][2], tdir=templates[int(item)-1][1])
            templates = templates_temp

        # print list of enabled templates
        self.display.print_list("TEMPLATE LIST", templates)
        
    #----------------------------
    # Load web sites
    #----------------------------
    def load_websites(self):
        # a required flags set?
        if self.config["enable_web"] == True:
            self.select_web_templates()
            print
            self.display.output("Starting phishing webserver")
            if (self.config["always_yes"] or self.display.yn("Continue", default="y")):

                path = os.path.dirname(os.path.realpath(__file__))
                # Start process
                cmd = [path + "/../web.py", Utils.compressDict(self.config)]
                self.webserver = subprocess.Popen(cmd, shell=False, stdout=subprocess.PIPE)

                # monitor output to gather website information
                while True:
                    line = self.webserver.stdout.readline()
                    line = line.strip()
                    if line == 'Websites loaded and launched.':
开发者ID:firebitsbr,项目名称:SPF,代码行数:70,代码来源:framework.py


注:本文中的display.Display.selectlist方法示例由纯净天空整理自Github/MSDocs等开源代码及文档管理平台,相关代码片段筛选自各路编程大神贡献的开源项目,源码版权归原作者所有,传播和使用请参考对应项目的License;未经允许,请勿转载。