mitrofun / weakorm

weekorm/model.py

from datetime import datetime

This module should have a docstring.

from weekorm import db


class Field:

This class should have a docstring.

The variable __class__ seems to be unused.

    field_type = ''
    default = ''
    is_foreign_key = False
    unique = False
    not_null = False
    python_type = str

    def field_sql(self, field_name):

This method should have a docstring.

        field_sql = f'"{field_name}" {self.field_type}'
        if self.unique:
            field_sql += ' UNIQUE'
        if self.not_null:
            field_sql += ' NOT NULL'
        return field_sql

    def to_python(self, value):

This method should have a docstring.

        if self.python_type == datetime:
            value = datetime.fromtimestamp(value)
            return value
        return self.python_type(value)


class CharField(Field):

This class should have a docstring.

The variable __class__ seems to be unused.

    def __init__(self, max_length=255, default='', unique=False):
        self.field_type = f'varchar({max_length})'
        self.default = default
        self.max_length = max_length
        self.unique = unique
        self.python_type = str


class IntegerField(Field):

This class should have a docstring.

The variable __class__ seems to be unused.

    def __init__(self, default=0, unique=False):
        self.field_type = 'integer'
        self.default = default
        self.unique = unique
        self.python_type = int


class FloatField(Field):

This class should have a docstring.

The variable __class__ seems to be unused.

    def __init__(self, default=0.0, unique=False):
        self.field_type = 'real'
        self.default = default
        self.unique = unique
        self.python_type = float


class BooleanField(Field):

This class should have a docstring.

The variable __class__ seems to be unused.

    def __init__(self, default=False, unique=False):
        self.field_type = 'integer'
        self.default = default
        self.unique = unique
        self.python_type = bool


class DateTimeField(Field):

This class should have a docstring.

The variable __class__ seems to be unused.

    def __init__(self, default=0.0, unique=False):
        self.field_type = 'real'
        self.default = default
        self.unique = unique
        self.python_type = datetime


class ForeignKey(Field):

This class should have a docstring.

The variable __class__ seems to be unused.

    def __init__(self, model_class):
        self.field_type = 'foreignkey'
        self.is_foreign_key = True
        self.model_class = model_class
        self.python_type = int

    def field_sql(self, field_name):
        return f'"{field_name}" integer REFERENCES "{self.model_class.__name__.lower()}" ("id")'


class Model:

This class should have a docstring.

The variable __class__ seems to be unused.

    def __init__(self, inst_id=0, **kwargs):
        self.table_name = self.__class__.__name__.lower()
        self.id = inst_id

The name id does not conform to the attribute naming conventions ((([a-z][a-z0-9_]{2,30})|(_[a-z0-9_]*))$).

        self.__class__.try_create_table()

        for name in self.field_names:
            field = getattr(self.__class__, name.replace("`", ""))
            setattr(self, name.replace("`", ""), field.default)
        for key, value in kwargs.items():
            setattr(self, key.replace("`", ""), value)

    def __str__(self):
        return f'{self.__class__.__name__}_{self.id}'

    def __repr__(self):
        return self.__str__()

    @property
    def field_names(self):

This method should have a docstring.

        names = []
        for name in dir(self.__class__):
            var = getattr(self.__class__, name.replace("`", ""))
            if isinstance(var, Field):
                names.append(f"`{name}`")
        return names

    @property
    def field_values(self):

This method should have a docstring.

        values = []
        for name in self.field_names:
            value = getattr(self, name.replace("`", ""))
            if isinstance(value, datetime):
                value = value.timestamp()
            if isinstance(value, Model):
                value = value.id
            if isinstance(value, str):
                value = value.replace("'", "''")
            values.append(f"'{value}'")
        return values

    def insert(self):

This method should have a docstring.

        cursor = db.get_cursor()
        field_names_sql = ", ".join(self.field_names)
        field_values_sql = ", ".join(self.field_values)

        sql = f"insert into `{self.table_name}`({ field_names_sql}) values({field_values_sql})"
        db.execute_sql(cursor, sql)
        db.db_commit()

        sql = f"select id from `{self.table_name}` order by id desc;"
        db.execute_sql(cursor, sql)
        self.id = cursor.fetchone()[0]

    def update(self):

This method should have a docstring.

        cursor = db.get_cursor()

        name_value = []

        for name, value in zip(self.field_names, self.field_values):
            name_value.append(f"{name}={value}")
        name_value_sql = ", ".join(name_value)

        sql = f"update `{self.table_name}` set {name_value_sql} where id = {self.id}"
        db.execute_sql(cursor, sql)
        db.db_commit()

    def save(self):

This method should have a docstring.

        if self.id:
            self.update()
        else:
            self.insert()
        return self

    def delete(self):

This method should have a docstring.

        cursor = db.get_cursor()
        sql = f"delete from `{self.table_name}` where id = {self.id}"
        db.execute_sql(cursor, sql)
        db.db_commit()

    @classmethod
    def try_create_table(cls):

This method should have a docstring.

        table_name = cls.__name__.lower()
        cursor = db.get_cursor()
        sql = f"select * from sqlite_master where type='table' AND name='{table_name}';"
        db.execute_sql(cursor, sql)
        if not cursor.fetchall():
            sql = f"drop table if exists `{table_name}`;"
            db.execute_sql(cursor, sql)

            fields_sql = ""
            for name in dir(cls):
                var = getattr(cls, name.replace("`", ""))
                if isinstance(var, Field):
                    field = var
                    field_sql = field.field_sql(name)
                    fields_sql += ", " + field_sql
            sql = f'create table `{table_name}` ("id" integer not null primary key {fields_sql});'
            db.execute_sql(cursor, sql)

            db.db_commit()

    @classmethod
    def query(cls):

This method should have a docstring.

        query = Query(cls)
        return query

    @classmethod
    def get(cls, **kwargs):

This method should have a docstring.

        query = Query(cls)
        return query.filter(**kwargs).first()


class Query:

This class should have a docstring.

The variable __class__ seems to be unused.

    def __init__(self, model_class):
        model_class.try_create_table()
        self.model_class = model_class
        self.table_name = self.model_class.__name__
        self.where_sql = "1=1"

    def __str__(self):
        return f"{self.__class__.__name__}_{self.table_name,}_{self.query_sql}"

    def __repr__(self):
        return self.__str__()

    @property
    def field_names(self):

This method should have a docstring.

        names = []
        for name in dir(self.model_class):
            var = getattr(self.model_class, name.replace("`", ""))
            if isinstance(var, Field):
                names.append(f"`{name}`")
        return names

    @property
    def query_sql(self):

This method should have a docstring.

        sql = f"select * from `{self.table_name}` where {self.where_sql};"
        return sql

    def filter(self, operator="=", **kwargs):

This method should have a docstring.

        where_sql = self.where_sql
        for name, value in kwargs.items():
            if f"`{name}`" in self.field_names + ["`id`"]:
                if isinstance(value, Model):
                    value = value.id
                if isinstance(value, str):
                    value = value.replace("'", "''")
                where_sql += f" and `{name}` {operator} '{value}'"
        query = self.__class__(self.model_class)
        query.where_sql = where_sql
        return query

    def _r2ob(self, item):
        value = ''
        inst_id = item[0]
        obj = self.model_class(inst_id=inst_id)
        for i in range(1, len(item)):
            name = self.field_names[i - 1]
            field = getattr(self.model_class, name.replace("`", ""))
            if not field.is_foreign_key:
                value = field.to_python(item[i])
            if field.is_foreign_key:
                if field.field_type == "foreignkey":
                    fid = item[i]
                    value = field.model_class.get(id=fid)
            setattr(obj, name.replace("`", ""), value)
        return obj

    def all(self):

This method should have a docstring.

        cursor = db.get_cursor()
        sql = self.query_sql
        db.execute_sql(cursor, sql)
        rows = cursor.fetchall()
        obs = []
        for item in rows:
            ob = self._r2ob(item)

The name ob does not conform to the variable naming conventions ((([a-z][a-z0-9_]{2,30})|(_[a-z0-9_]*))$).

            obs.append(ob)
        return obs

    def first(self):

This method should have a docstring.

        cursor = db.get_cursor()
        sql = self.query_sql
        db.execute_sql(cursor, sql)
        rows = cursor.fetchall()
        if rows:

Unnecessary "else" after "return"

            r = rows[0]

The name r does not conform to the variable naming conventions ((([a-z][a-z0-9_]{2,30})|(_[a-z0-9_]*))$).

            ob = self._r2ob(r)

The name ob does not conform to the variable naming conventions ((([a-z][a-z0-9_]{2,30})|(_[a-z0-9_]*))$).

            return ob
        else:
            return None

    def delete(self):

This method should have a docstring.

        cursor = db.get_cursor()
        sql = f"delete from `{self.table_name}` where {self.where_sql}"
        db.execute_sql(cursor, sql)
        db.db_commit()