CellarTracker only bottle-list __init__.py

# replace "__init__.py" from https://github.com/ahoernecke/ha_cellar_tracker to import all bottles in Home Assistant

from cellartracker import cellartracker
import pandas as pd
import numpy as np
import logging

from random import seed
from random import randint
from datetime import timedelta

from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
import voluptuous as vol
import homeassistant.helpers.config_validation as cv
from homeassistant.util import Throttle


"""Example Load Platform integration."""
DOMAIN = 'cellar_tracker'

SCAN_INTERVAL = timedelta(seconds=3600)
MIN_TIME_BETWEEN_UPDATES = timedelta(seconds=3600)

_LOGGER = logging.getLogger(__name__)

CONFIG_SCHEMA = vol.Schema(
    {
        DOMAIN: vol.Schema(
            {
                vol.Required(CONF_USERNAME): cv.string,
                vol.Required(CONF_PASSWORD): cv.string,
            }
        )
    },
    extra=vol.ALLOW_EXTRA,
)


def setup(hass, config):
    """Your controller/hub specific code."""
    # Data that you want to share with your platforms

    conf = config[DOMAIN]

    username = conf[CONF_USERNAME]
    password = conf[CONF_PASSWORD]

    hass.data[DOMAIN] = WineCellarData(username, password)
    hass.data[DOMAIN].update()

    hass.helpers.discovery.load_platform('sensor', DOMAIN, {}, config)


    return True

class WineCellarData:
    """Get the latest data and update the states."""

    def __init__(self, username, password):
        """Init the Canary data object."""

        self._username = username
        self._password = password

    def get_reading(self, key):
      return self._data[key]

    def get_readings(self):
      return self._data

    @Throttle(MIN_TIME_BETWEEN_UPDATES)
    def update(self, **kwargs):
      data = {}
      username = self._username
      password = self._password

      client = cellartracker.CellarTracker(username, password)

      list = client.get_list()
      df = pd.DataFrame(list)
      columns = ["Quantity", "Type", "Country", "Region", "Vintage", "Varietal", "Wine", "BeginConsume", "EndConsume", "CT", "Size", "Price", "Location", "Bin"] #define columns
      group_data = df.reindex(columns=columns) #get columns
      group_data[["Quantity"]] = group_data[["Quantity"]].apply(pd.to_numeric) #make Quantity a numeric value
      group_data.columns = ['count', 'type', 'country', "region", "vintage", "varietal", "wine", "beginconsume", "endconsume", "score", "size", "price", "location", "bin"] #rename columns
      group_data.index = df['iWine'] #create index (unique name)

      group='Flaske' #Add additional data to dict
      data[group] = {}
      for row, item in group_data.iterrows():
          data[group][row] = item.to_dict()
          data[group][row]["sub_type"] = row

      data["total_bottles"] = group_data['count'].sum()

      self._data = data



#Headers can be found here (remember to login to Cellartracker prior to):
#https://www.cellartracker.com/xlquery.asp?Table=list
#see also https://support.cellartracker.com/article/29-exporting-data