base_model.py

#!/usr/bin/python3
"""
Contains class BaseModel
"""

from datetime import datetime
import models
from os import getenv
import sqlalchemy
from sqlalchemy import Column, String, DateTime
from sqlalchemy.ext.declarative import declarative_base
import uuid

time = "%Y-%m-%dT%H:%M:%S.%f"

if models.storage_t == "db":
    Base = declarative_base()
else:
    Base = object


class BaseModel:
    """The BaseModel class from which future classes will be derived"""
    if models.storage_t == "db":
        id = Column(String(60), primary_key=True)
        created_at = Column(DateTime, default=datetime.utcnow)
        updated_at = Column(DateTime, default=datetime.utcnow)

    def __init__(self, *args, **kwargs):
        """Initialization of the base model"""
        if kwargs:
            for key, value in kwargs.items():
                if key in ('created_at', 'updated_at'):
                    setattr(self, key, datetime.strptime(
                        value, "%Y-%m-%dT%H:%M:%S.%f"))
                elif key == '__class__':
                    continue
                else:
                    setattr(self, key, value)
        else:
            self.id = str(uuid.uuid4())
            self.created_at = datetime.now()
            self.updated_at = self.created_at
        # if kwargs:
        #     for key, value in kwargs.items():
        #         if key in ("created_at", "updated_at"):
        #             setattr(self, key, datetime.strptime(
        #                 value, '%Y-%m-%dT%H:%M:%S.%f'))
        #         elif key == '__class__':
        #             continue
        #         else:
        #             setattr(self, key, value)
        #     if not self.id:
        #         setattr(self, 'id', str(uuid.uuid4()))
        #     now = datetime.utcnow()
        #     if not self.created_at:
        #         setattr(self, 'created_at', now)
        #     if not self.updated_at:
        #         setattr(self, 'updated_at', now)
        # else:
        #     self.id = str(uuid.uuid4())
        #     self.created_at = self.updated_at = datetime.utcnow()
        # if kwargs:
        #     for key, value in kwargs.items():
        #         if key != "__class__":
        #             setattr(self, key, value)
            # if kwargs.get("created_at", None) and type(self.created_at)
            # is str:
            # if kwargs.get("created_at", None) and isinstance(self.created_at,
            #                                                  str):
            #     self.created_at = datetime.strptime(
            #         kwargs["created_at"], time)
        #     else:
        #         self.created_at = datetime.utcnow()
        #     if kwargs.get("updated_at", None) and type(self.updated_at)
            # is str:
            # if kwargs.get("updated_at", None) and isinstance(self.updated_at,
            #                                                  str):
            #     self.updated_at = datetime.strptime(
            #       kwargs["updated_at"], time)
        #     else:
        #         self.updated_at = datetime.utcnow()
        #     if kwargs.get("id", None) is None:
        #         self.id = str(uuid.uuid4())
        # else:
        #     self.id = str(uuid.uuid4())
        #     self.created_at = datetime.utcnow()
        #     self.updated_at = self.created_at

    def __str__(self):
        """String representation of the BaseModel class"""
        return "[{:s}] ({:s}) {}".format(self.__class__.__name__, self.id,
                                         self.__dict__)

    def save(self):
        """updates the attribute 'updated_at' with the current datetime"""
        self.updated_at = datetime.utcnow()
        models.storage.new(self)
        models.storage.save()

    def to_dict(self):
        """returns a dictionary containing all keys/values of the instance"""
        new_dict = self.__dict__.copy()
        if "created_at" in new_dict:
            new_dict["created_at"] = new_dict["created_at"].strftime(time)
        if "updated_at" in new_dict:
            new_dict["updated_at"] = new_dict["updated_at"].strftime(time)
        new_dict["__class__"] = self.__class__.__name__
        if "_sa_instance_state" in new_dict:
            del new_dict["_sa_instance_state"]
        return new_dict

    def delete(self):
        """delete the current instance from the storage"""
        models.storage.delete(self)