写一个简单的ORM

最近在研究SQLAlchemy源码,自己造个轮子先(当然,SQLAlchemy远比这里复杂)

import logging

logging.basicConfig(level=logging.DEBUG)


class Column:
    def __init__(self, type, default=None):
        self.col_name = None
        self.col_type = type
        self.col_default = default

    def __eq__(self, value):
        logging.debug("__eq__ of {} && {}".format(self, value))
        return "{} = {}".format(self.col_name, value)


class Base(type):
    def __new__(cls, *args, **kwargs):
        logging.debug("creating class {}, args {}, kwargs {}".format(
            cls, args, kwargs
        ))

        if len(args) > 2:
            for k, v in args[2].items():
                if k.startswith("_"):
                    continue

                v.col_name = k

        return type.__new__(cls, *args, **kwargs)


class User(metaclass=Base):
    __tablename__ = "user"

    id = Column(int)
    name = Column(str)
    passwd = Column(str)


class Queryable:
    def __init__(self, table):
        self.table = table

    def filter(self, text):
        return "select * from {} where {}".format(
            self.table.__tablename__,
            text,
        )


class Session:
    def query(self, table):
        logging.debug("returning Queryable({})".format(table))
        return Queryable(table)


session = Session()
print(session.query(User).filter(User.id == 1))

运行结果:

jiajun@debian test: python3 orm.py
DEBUG:root:creating class <class '__main__.Base'>, args ('User', (), {'__qualname__': 'User', '__tablename__': 'user', 'id': <__main__.Column object at 0x7f8174b107b8>, 'passwd': <__main__.Column object at 0x7f8174b10940>, 'name': <__main__.Column object at 0x7f8174b10828>, '__module__': '__main__'}), kwargs {}
DEBUG:root:returning Queryable(<class '__main__.User'>)
DEBUG:root:__eq__ of <__main__.Column object at 0x7f8174b107b8> && 1
select * from user where id = 1

微信公众号
关注公众号,获得及时更新

更多文章
  • Go设计模式:适配器模式
  • Go设计模式:Iterator
  • glusterfs 笔记
  • 用peewee代替SQLAlchemy
  • Go的slice工作机制
  • Linux系统迁移记录(从HDD到SSD)
  • Redis是如何工作的?
  • virsh自动关闭windows虚拟机
  • Golang sort源码阅读
  • 分治的思维方式
  • Debian 使用NetworkManager之后networking.service崩溃
  • httprouter源码阅读与分析
  • 《程序员的自我修养-装载、链接与库》笔记
  • Golang sync.Pool源码阅读与分析
  • MySQL操作笔记