def get(self, id):
try:
query = models.Post.get(models.Post.id == id)
post_schema = models.PostSchema(only=('id', 'content', 'title',
'author.name', 'author.id',
'is_url', 'created_at',
'last_modified'))
post = post_schema.dump(query).data
print(post)
query = (models.Tag.select(models.Tag).join(
models.PostTags,
JOIN.RIGHT_OUTER).where(models.PostTags.post == id))
tag_schema = models.TagSchema(many=True)
tags = tag_schema.dump(query).data
post['tags'] = tags
return jsonify({'post': post})
except models.DoesNotExist:
abort(404, message="Record does not exist.")
def get(self, name):
try:
query = models.Tag.get(models.Tag.name == name)
tag_schema = models.TagSchema()
output = tag_schema.dump(query).data
return jsonify({'tag': output})
except models.DoesNotExist:
return jsonify({'error': {'message': 'record does not exist.'}})
def get(self):
try:
query = models.Tag.select().order_by(models.Tag.id)
tag_schema = models.TagSchema(many=True)
output = tag_schema.dump(query).data
return jsonify({'tags': output})
except:
pass
def get(self, id):
try:
query = (models.Tag.select(models.Tag).join(
models.PostTags,
JOIN.RIGHT_OUTER).where(models.PostTags.post == id))
except:
abort(404, message="Record does not exist.")
tag_schema = models.TagSchema(many=True)
output = tag_schema.dump(query).data
return jsonify({'tags': output})
def copy_tags(iSourceUUID, iDestinationUUID):
new_tags = []
session = Session()
try:
temp_dict = session.query(md.Tag). \
filter(md.Tag.uuid == iSourceUUID). \
all()
tag_dict = md.TagSchema(many=True).dump(temp_dict)
for row in tag_dict:
if gv._DEBUG:
print("f(x) copy_tags: VALUES BEFORE COPY: {}:{}:{}:{}".format(
row["uuid"], row["galaxy"], row["tag"], row["type"]))
newRecord = (iDestinationUUID, row["galaxy"], row["tag"],
row["type"])
new_tags.append(newRecord)
return new_tags
except Exception as error:
print("f(x) get_parent_child_data: DATABASE ERROR: {}".format(error))
sys.exit(error)
finally:
session.close()
def get_set_all_tags(iUUID):
ret_list = []
session = Session()
try:
# query = "SELECT * FROM mp_tags WHERE uuid = ?"
tmp_dict = session.query(md.Tag). \
filter(md.Tag.uuid == iUUID). \
all()
tag_dict = md.TagSchema(many=True).dump(tmp_dict)
# BUILD RETURN LIST
for tag in tag_dict:
tmpGalaxy = tag["galaxy"]
tmpTagVal = tag["tag"]
tmpTypeTag = tag["type"]
newTag = MISPTag()
# POTENTIAL tmpTypeTag Values
# "ACTOR"
# "COUNTRY_SPONSOR"
# "GALAXY"
# "GALAXY_SYNONIM"
# "ISO_COUNTRY"
# "MALWARE"
# "TARGETS"
# "TYPE_OF_INCIDENT"
# "VICTIMS"
if "tlp:" in tmpTagVal:
newTag.name = tmpTagVal
elif tmpTypeTag == "ACTOR":
newTag.name = "Actor: " + tmpTagVal
newTag.colour = gv._ACTOR_TAG
elif tmpTypeTag == "COUNTRY_SPONSOR":
newTag.name = "Sponsor: " + tmpTagVal
newTag.colour = gv._COUNTRY_SPONSOR_TAG
elif tmpTypeTag == "GALAXY":
if tmpGalaxy != "":
newTag.name = "misp-galaxy:" + tmpGalaxy + "=\"" + tmpTagVal + "\""
else:
newTag.name = "Synonym: " + tmpTagVal
newTag.colour = gv._GALAXY_SYNONIM_TAG
elif tmpTypeTag == "GALAXY_SYNONIM":
newTag.name = "Synonym: " + tmpTagVal
newTag.colour = gv._GALAXY_SYNONIM_TAG
elif tmpTypeTag == "ISO_COUNTRY":
newTag.name = "Country ISO: " + tmpTagVal
newTag.colour = gv._ISO_COUNTRY_TAG
elif tmpTypeTag == "MALWARE":
newTag.name = "Malware: " + tmpTagVal
newTag.colour = gv._MALWARE_TAG
elif tmpTypeTag == "TARGETS":
newTag.name = "Target: " + tmpTagVal
newTag.colour = gv._TARGETS_TAG
elif tmpTypeTag == "TYPE_OF_INCIDENT":
newTag.name = "Type of Incident: " + tmpTagVal
newTag.colour = gv._TYPE_OF_INCIDENT_TAG
elif tmpTypeTag == "VICTIMS":
newTag.name = "Victim: " + tmpTagVal
newTag.colour = gv._VICTIMS_TAG
else:
newTag.name = tmpTagVal
newTag.colour = gv._OTHER_TAG
ret_list.append(newTag)
return ret_list
except Exception as error:
print("f(x) get_set_all_tags: DATABASE ERROR: {}".format(error))
sys.exit(error)
finally:
session.close()