import sys
from pytaxize import col

class Ids(object):
    '''
    ids: A class for taxonomic identifiers

    Usage::

      import pytaxize

      res = pytaxize.Ids('Poa annua', db='col')
      res.get_colid()
    '''
    def __init__(self, name, db):
        # super(ids, self).__init__()
        self.db = db
        self.name = name

    def __repr__(self):
      return """<%s %s:%s>""" % (type(self).__name__, self.db, self.name)

    # def get_colid(sciname=self.sciname, ask = True, verbose = True):
    def get_colid(self, ask = True, verbose = True):
      '''
      pytaxize.get_colid(sciname=['Poa annua'])
      '''
      sciname = self.name
      def fun(sciname, ask, verbose):
        sciname = [sciname]
        df = col.col_search(name=sciname)

        if(df[0].shape[0] == 0):
          sys.exit("Retrieving data for taxon '" + sciname + "'")
          id = 'none'
        else:
          df = df[0][['id','name','rank','name_status']]
          df.columns = ['colid', 'name', 'rank', 'name_status']
          id = df['colid'].values.tolist()

        # not found on col
        if(len(id) == 0):
          sys.exit("Not found. Consider checking the spelling or alternate classification")
          id = 'none'

        # more than one found on col -> user input
        if(len(id) > 1):
            if(ask==True):
                print("\nMore than one eolid found for taxon '" + sciname[0] + "'\n")
                print(df)
                take = input("\n Enter rownumber of taxon:\n\n")

                if(len(str(take)) == 0):
                    take = 'notake'
                else:
                    pass
                if(int(take) in range(df.shape[0])):
                    take = int(take)
                    print("Input accepted, took eolid '" + df['colid'][take] + "'.\n")
                    id = int(df['colid'][take])
                else:
                    id = 'none'
                    print("\nReturned 'none'!\n\n")
            else:
                id = 'none'
        return id

      sciname = converter(sciname)
      out = []
      for i in range(len(sciname)):
        out.append(fun(sciname[i], ask, verbose))

      return out

    def getcolurl(self):
      return self.name + 'Boo'

def converter(x):
    if(x.__class__.__name__ == 'str'):
        return [x]
    else:
        return x