processors.utils

Complexity

Total Complexity

5

Size/Duplication

Total Lines	54
Duplicated Lines	0 %

Importance

Changes

0

3 Functions

Rating	Name	Duplication	Size	Complexity
A	is_string()	0	2	1
A	full_path()	0	5	1
A	post_json()	0	15	1

# -*- coding: utf-8 -*-
from __future__ import unicode_literals
from termcolor import colored
import requests
import json
import os

def is_string(x):
    return isinstance(x, ("".__class__, u"".__class__))

def post_json(service, json_data):
    # POST json to the server API
    #response = requests.post(service, json={"text":"{}".format(text)})
    # for older versions of requests, use the call below
    #print("SERVICE: {}".format(service))
    response = requests.post(service,
                             data=json_data,
                             headers={'content-type': 'application/json; charset=utf-8'},
                             timeout=None
                             )
    # response content should be utf-8
    #response.encoding = "utf-8"
    content = response.content.decode("utf-8")
    #print("CONTENT: {}".format(content))
    return json.loads(content)

def full_path(p):
    """
    Expand a path.  Supports "~" shortcut.
    """
    return os.path.abspath(os.path.normpath(os.path.expanduser(p)))

class LabelManager(object):
    """
    Keep track of common labels
    """
    UNKNOWN = "UNKNOWN"
    # the O in IOB notation
    O = "O"


class OdinError(Exception):
    """
    An error encountered while parsing an Odin rule.
    """

    def __init__(self, rules, message):
        self.rules = rules
        self.message = message

    def __str__(self):
        cn = colored(self.__class__.__name__, color="red", attrs=["bold"])
        return "{}: {}\n{}".format(cn, self.message, self.rules)