MISP
/
PyMISPGalaxies
mirror of https://github.com/MISP/PyMISPGalaxies
2
0
Fork 0
Code Issues Releases Wiki Activity

Add revert machine tag method

pull/3/head
Raphaël Vinot 2017-07-25 18:57:18 +02:00
parent 2cf0fc2d30
commit 27a2f8e7d9
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
pymispgalaxies/api.py
View File

@ -7,6 +7,7 @@ import os
import sys import sys
import collections import collections
from glob import glob from glob import glob
import re
try: try:
import jsonschema import jsonschema
@ -157,6 +158,10 @@ class ClusterValue():
to_return['meta'] = self.meta._json() to_return['meta'] = self.meta._json()
return to_return return to_return
def __str__(self):
# TODO: improve that
return '{}\n{}'.format(self.value, self.description)
class Cluster(): class Cluster():
@ -200,7 +205,7 @@ class Clusters(collections.Mapping):
for cluster_file in glob(os.path.join(self.root_dir_clusters, '*.json')): for cluster_file in glob(os.path.join(self.root_dir_clusters, '*.json')):
with open(cluster_file, 'r') as f: with open(cluster_file, 'r') as f:
cluster = json.load(f) cluster = json.load(f)
self.clusters[cluster['name']] = Cluster(cluster) self.clusters[cluster['type']] = Cluster(cluster)
def validate_with_schema(self): def validate_with_schema(self):
if not HAS_JSONSCHEMA: if not HAS_JSONSCHEMA:
@ -215,6 +220,14 @@ class Clusters(collections.Mapping):
def all_machinetags(self): def all_machinetags(self):
return [cluster.machinetags() for cluster in self.clusters.values()] return [cluster.machinetags() for cluster in self.clusters.values()]
def revert_machinetag(self, machinetag):
_, cluster_type, cluster_value = re.findall('^([^:]*):([^=]*)="([^"]*)"$', machinetag)[0]
cluster = self.clusters[cluster_type]
for v in cluster.values:
if v.value == cluster_value:
return cluster, v
return None
def __getitem__(self, name): def __getitem__(self, name):
return self.clusters[name] return self.clusters[name]