import orthancInterface
import json
import chardet

def extractJSON(data):
    encoding=chardet.detect(data)["encoding"]
    return json.loads(data.decode(encoding))


class orthancDB:
    def __init__(self,net):
        self.net=net

    def getList(self,category):
        url=self.net.getCoreURL()
        url+='/'+category
        response=self.net.get(url)
        return extractJSON(response.data)
    
    def getPatients(self):
        return self.getList('patients')

    def getStudies(self):
        return self.getList('studies')
    
    def getPatientData(self, orthancId):
        return self.getData('patients',orthancId)

    def getStudyData(self, studyOrthancId):
        return self.getData('studies',studyOrthancId)

    def getSeriesData(self, seriesOrthancId):
        return self.getData('series',seriesOrthancId)

    def getData(self, level, orthancId):
        url=self.net.getCoreURL()
        url+='/'+level+'/'+orthancId
        response=self.net.get(url)
        return extractJSON(response.data)


    def getDicomField(self,orthancInstanceId,dicomField):
        url=self.net.getCoreURL()
        url+='/instances/'+orthancInstanceId+'/content/'+dicomField
        response=self.net.get(url)
        try:
            encoding=chardet.detect(response.data)["encoding"]
            return response.data.decode(encoding)
        except TypeError:
            return ''

    def selectRows(self, qfilter):
        obj={}
        obj["Level"]="Instance"
        obj["Query"]=qfilter
        jsonData=json.dumps(obj)

        url=self.net.getCoreURL()
        url+='/tools/find'
        response=self.net.post(url,'json',jsonData)
        encoding=chardet.detect(response.data)["encoding"]
        return json.loads(response.data.decode(encoding))


    def upload(self, f) :
        url=self.net.getCoreURL()
        url+='/instances'
        fobj=open (f, 'rb')
        response=self.net.post(url,'octet-stream',fobj.read())
        try:
            encoding=chardet.detect(response.data)["encoding"]
        except AttributeError:
            print(response)
            return json.loads('{"status":"BAD"}')
        return json.loads(response.data.decode(encoding))

    def remove(self,level,orthancId):
        url=self.net.getCoreURL()
        url+='/'+level+'/'+orthancId
        response=self.net.delete(url)
        print(response.data)