示例#1
0
文件: fetch.py 项目: jasonish/ruleman
def main(args):
    usage = """
usage: ruleman fetch [options] [ruleset0 ruleset1 ...]

Options:

    -f, --force

If no ruleset names are specified the only those enabled will be
fetched, otherwise only the specified rulesets will be fetched even if
not enabld.
"""

    force = False

    try:
        opts, args = getopt.getopt(args, "hf", ["help", "force"])
    except getopt.GetoptError as err:
        print(usage, file=sys.stderr)
        return 1
    for o, a in opts:
        if o in ["-h", "--help"]:
            print(usage)
            return 0
        elif o in ["-f", "--force"]:
            force = True

    if not os.path.exists(config.RULESET_DATA_DIR):
        print("Creating directory %s." % config.RULESET_DATA_DIR)
        os.makedirs(config.RULESET_DATA_DIR)

    rulesets = config.get_rulesets()
    fetched = []
    for ruleset in rulesets.values():
        ret = False
        if args:
            if ruleset["name"] in args:
                ret = fetch_ruleset(ruleset, force=force)
        else:
            ret = fetch_ruleset(ruleset, force=force)
        if ret:
            fetched.append(ruleset["name"])

    # Return a list of ruleset names that were actually fetched.
    return fetched
示例#2
0
文件: main.py 项目: jasonish/ruleman
    def search(self):

        rulesets = config.get_rulesets()
        for ruleset in rulesets.itervalues():
            core.load_ruleset_files(ruleset)
            core.load_ruleset_rules(ruleset)
            logger.info("Loaded ruleset %s: %d files; %d rules" % (
                    ruleset["name"], len(ruleset["files"]), 
                    len(ruleset["rules"])))

        match_count = 0
        for ruleset in rulesets.itervalues():
            for rule in ruleset["rules"].itervalues():
                if self.is_match(rule):
                    match_count += 1
                    self.print_rule(ruleset["name"], rule)

        print("Found %d rules." % (match_count))
示例#3
0
文件: test.py 项目: jasonish/ruleman
 def test_get_rulesets(self):
     rulesets = config.get_rulesets()
     self.assertTrue("partial" in rulesets)
     self.assertTrue("complete" in rulesets)