Source code for synapse.data

import os
import logging

import synapse.common as s_common

import synapse.lib.json as s_json
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 s_json.load(fd)




def getLark(name):
    with s_datfile.openDatFile(f'synapse.data.lark/{name}.lark') as larkf:
        return larkf.read().decode()




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