import labkeyInterface
import json
import chardet

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

   def selectRows(self,project,schemaName, queryName,qfilter,viewName='default'):

      debug=False
      url=self.net.GetLabkeyUrl()+'/'+project
      url+='/query-selectRows.api?schemaName='+schemaName+\
                '&query.queryName='+queryName
      if viewName!='default':
         url+='&query.viewName={}'.format(viewName)
      for f in qfilter:
         url+="&query."+f['variable']+"~"+f['oper']+"="+f['value']
      if debug:
         print("Sending {}").format(url)
      response=self.net.get(url)
      encoding=chardet.detect(response.data)["encoding"]
      return json.loads(response.data.decode(encoding))

   def modifyRows(self,mode, project,schemaName, queryName, rows):
      #mode can be insert/update/delete
      debug=True
      data={}
      data['schemaName']=schemaName
      data['queryName']=queryName
      data['rows']=rows
      url=self.net.GetLabkeyUrl()+'/'+project
      url+='/query-'+mode+'Rows.api?'
      response=self.net.post(url,json.dumps(data))
      return response.data

   def addQuery(self,project,schemaName,queryName,fields):
      listDomainDefinition={}
      if schemaName=='lists':
         listDomainDefinition['kind']='IntList'
         fields.insert(0,{'name':'Key','rangeURI':'int'})
         options={}
         options['keyName']='Key'
         if queryName.find('enum')==0:
            options['keyType']='Integer'
         else:
            options['keyType']='AutoIncrementInteger'
         listDomainDefinition['options']=options
      if schemaName=='study':
         listDomainDefinition['kind']='StudyDatasetVisit'
      domainDesign={}
      domainDesign['name']=queryName
      domainDesign['description']='Some description'
      domainDesign['fields']=fields
      listDomainDefinition['domainDesign']=domainDesign
      url=self.net.GetLabkeyUrl()+'/'+project
      url+='/property-createDomain.api'
      response=self.net.post(url,json.dumps(listDomainDefinition))
      return response.data

   def getQueryDesign(self,project,schemaName,queryName):
      url=self.net.GetLabkeyUrl()+'/'+project
      url+='/property-getDomain.api?'
      url+='schemaName={}'.format(schemaName)
      url+='&queryName={}'.format(queryName)
      response=self.net.get(url)
      encoding=chardet.detect(response.data)["encoding"]
      return json.loads(response.data.decode(encoding))