from ktbs_bench.graph_store import GraphStore
import rdflib

rdflib.plugin.register('BN', rdflib.store.Store, 'ktbs_bench.bnsparqlstore', 'SPARQLUpdateStore')


def get_sparqlstore(query_endpoint, update_endpoint, identifier="http://localhost/generic_sparqlstore/"):
    triple_store = GraphStore(store='BN', identifier=identifier,
                              connect_args={'configuration': (query_endpoint, update_endpoint)})
    return triple_store

Add implementation of more generic store calls