Source code for synapse.data

import os
import json
import logging

import synapse.common as s_common
import synapse.lib.datfile as s_datfile
import synapse.lib.msgpack as s_msgpack

logger = logging.getLogger(__name__)

dirname = os.path.dirname(__file__)

def get(name, defval=None):
    '''
    Return an object from the embedded synapse data folder.

    Example:

        for tld in synapse.data.get('iana.tlds'):
            dostuff(tld)

    NOTE: Files are named synapse/data/<name>.mpk
    '''
    with s_datfile.openDatFile(f'synapse.data/{name}.mpk') as fd:
        return s_msgpack.un(fd.read())

def getJSON(name):
    with s_datfile.openDatFile(f'synapse.data/{name}.json') as fd:
        return json.loads(fd.read())

def path(*names):
    return s_common.genpath(dirname, *names)