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
|
import json
import click
from .. import mongo
from ..common.mongo import collection_status
from . import pp
from .mongo import create as create_collection
from .mongo import drop as drop_collection
from .mongo import query as query_collection
from .tasks import update_kb as update_kb_core
@pp.cli.command("create", help="Create the DB of protection profiles.")
def create(): # pragma: no cover
create_collection()
@pp.cli.command("drop", help="Drop the DB of protection profiles.")
def drop(): # pragma: no cover
drop_collection()
@pp.cli.command("query", help="Query the MongoDB for protection profiles.")
@click.option("-p", "--projection", type=json.loads, help="Projection to use with the query.")
@click.argument("query", type=json.loads)
def query(query, projection): # pragma: no cover
docs = query_collection(query, projection)
for doc in docs:
print(json.dumps(doc, indent=2))
@pp.cli.command("status", help="Print status information for the MongoDB collection.")
def status(): # pragma: no cover
collection_status(mongo.db.pp)
@pp.cli.command("update-kb", help="Update the KB of PP certs.")
def update_kb():
ids = list(map(lambda doc: doc["_id"], mongo.db.pp.find({}, {"_id": 1})))
reports = [(dgst, "report", None) for dgst in ids]
targets = [(dgst, "target", None) for dgst in ids]
update_kb_core(reports + targets)
|
