D-Net

from pydantic import BaseModel, Schema

from typing import Dict, List

from datetime import datetime

from fastapi import HTTPException

import logging

from time import time

class LinkProvider(BaseModel):

    name:str = Schema(None, title= "The name of the Provider that provides the links", max_length= 300)

    totalRelationships:int=Schema(None, title= "The number of links that It provides")

class LinkPublisher(BaseModel):

    name:str = Schema(None, title= "The name of the Publisher that provides the links", max_length= 300)

    totalRelationships:int=Schema(None, title= "The number of links that It provides")

class IdentifierType(BaseModel):

    identifier: str = None

    schema_str:str = Schema(None, alias="schema")

class ScholixProviderType(BaseModel):

    name:str

    identifiers:List[IdentifierType]  =None

class RelationshipType(BaseModel):

    name:str

    schema_str:str = Schema(None, alias="schema")

    inverseRelationship:str = None

class CreatorType(BaseModel):

    name: str

    identifiers:List[IdentifierType] = None

class ScholixObjectType(BaseModel):

    subtype:str = None

    type:str = None

class ScholixItemType(BaseModel):

    identifiers:List[IdentifierType] =None

    title:str = None

    objectType:str 

    creators:List[CreatorType] =None    

    publisher:List[ScholixProviderType] =None 

    objectProvider: List[ScholixProviderType] =None  

class ScholixType(BaseModel):

    linkProvider:List[ScholixProviderType] =None    

    publicationDate:str =None

    relationship:RelationshipType=None

    source:ScholixItemType=None

    target:ScholixItemType=None

def convert_response(response):

    log = logging.getLogger("scholexplorer")    

    for item in response.hits:

        result = item.__dict__['_d_']        

        result['linkProvider'] = result.pop('linkprovider')

        if 'creator' in result['source']:

            result['source']['creators']=result['source'].pop('creator')        

        result['source'].pop('objectSubType')

        result['source']['identifiers']=result['source'].pop('identifier')

        result['source']['objectProvider'] = [s['provider'] for s in result['source'].get('collectedFrom',[])]

        if result['source']['publisher'] is not None:

            result['source']['publisher'] = [x for x in result['source']['publisher'] if x.get('name') is not None]

        if 'creator' in result['target']:

            result['target']['creators']=result['target'].pop('creator')

        result['target'].pop('objectSubType')

        result['target']['identifiers']=result['target'].pop('identifier')

        if result['target'].get('publisher') is not None:

            result['target']['publisher'] = [x for x in result['target']['publisher'] if x.get('name') is not None]

        if result['target'].get('collectedFrom') is not None:

            result['target']['objectProvider'] = [s['provider'] for s in result['target'].get('collectedFrom',[])]

        else:

            result['target']['objectProvider'] = []

        yield  result