U
S��W� � @ s d Z ddlT G dd� de�ZdS )ag
Support for using a dedicated read-slave. The read database is specified as a
Model.Meta option, and will be used for SELECT statements:
master = PostgresqlDatabase('master')
read_slave = PostgresqlDatabase('read_slave')
class BaseModel(ReadSlaveModel):
class Meta:
database = master
read_slaves = [read_slave] # This database will be used for SELECTs.
# Now define your models as you would normally.
class User(BaseModel):
username = CharField()
# To force a SELECT on the master database, you can instantiate the SelectQuery
# by hand:
master_select = SelectQuery(User).where(...)
� )�*c s< e Zd Zedd� �Ze� fdd��Ze� fdd��Z� ZS )�ReadSlaveModelc C sF t | jdd �s| jjS t | dd�}|d t| jj� | _| jj| j S )N� read_slaves�_read_slave_idx���� )�getattrZ_meta�database�lenr r )�clsZ current_idx� r �/read_slave.py�_get_read_database s
z!ReadSlaveModel._get_read_databasec s t t| �j||�}| �� |_|S )N)�superr �selectr r �r �args�kwargsZquery�� __class__r r
r # s
zReadSlaveModel.selectc s0 t t| �j||�}|j�� �d�r,| �� |_|S )Nr )r r �rawZ_sql�lower�
startswithr r r r r r
r ) s
zReadSlaveModel.raw)�__name__�
__module__� __qualname__� classmethodr r r �
__classcell__r r r r
r s
r N)�__doc__ZpeeweeZModelr r r r r
�<module> s