#  Licensed to the Apache Software Foundation (ASF) under one
#  or more contributor license agreements.  See the NOTICE file
#  distributed with this work for additional information
#  regarding copyright ownership.  The ASF licenses this file
#  to you under the Apache License, Version 2.0 (the
#  "License"); you may not use this file except in compliance
#  with the License.  You may obtain a copy of the License at
#
#  http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing,
#  software distributed under the License is distributed on an
#  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
#  KIND, either express or implied.  See the License for the
#  specific language governing permissions and limitations
#  under the License.
from __future__ import annotations

from typing import Any, Dict

default_pandas_data_loader_config = {
    "if_exists": "replace",
    "chunksize": 500,
    "index": False,
    "method": "multi",
    "strftime": "%Y-%m-%d %H:%M:%S",
    "support_datetime_type": False,
}


class PandasLoaderConfigurations:
    if_exists: str
    chunksize: int
    index: bool
    method: str
    strftime: str
    support_datetime_type: bool

    def __init__(
        self,
        *,
        if_exists: str,
        chunksize: int,
        index: bool,
        method: str,
        strftime: str,
        support_datetime_type: bool,
    ):
        self.if_exists = if_exists
        self.chunksize = chunksize
        self.index = index
        self.method = method
        self.strftime = strftime
        self.support_datetime_type = support_datetime_type

    @classmethod
    def make_from_dict(cls, _dict: Dict[str, Any]) -> PandasLoaderConfigurations:
        copy_dict = default_pandas_data_loader_config.copy()
        copy_dict.update(_dict)
        return PandasLoaderConfigurations(**copy_dict)  # type: ignore

    @classmethod
    def make_default(cls) -> PandasLoaderConfigurations:
        return cls.make_from_dict({})