Расширение для Flask и sqlalchemy?

Question

[SilentSokolov]

Вообщем, возможно ли создать расширение для фласка аля Django-модуль, то есть которое может полноценно работать с sqlalchemy (создавать таблицу, сохранять)?

from flask.ext.blog import Blog()<br>
    blog = blog(app, db_session=session)<br>

Смысл в том, чтобы он работал с той же сессии, что и остальное приложение.



Есть другой вариант, создавать внутри приложения, и подсасываться в нему:

db = SQLAlchemy()<br>
<br>
    User(db.Model):<br>
    __tablename__ = 'users'<br>
    ....<br>

Подобное можно увидеть в этом расширении



Но тогда, если я захочу создать отношение (relationship) на модель в расширении, получу ошибку о том, что сессии разные и создать отношение невозможно.

db = SQLAlchemy() # новый!<br>
<br>
    Group(db.Model):<br>
    __tablename__ = 'groups'<br>
    ....<br>
    users = db.relationship('User') # ошибка мапинга<br>

Идеи?

Answer 1

[ramzes_yan]

а если db импортировать из User?

Comments

[SilentSokolov]

Не выйдет, тогда у тебя Group будет относиться к тоже сессии, что и User, теперь добавим еще один relationship на модель не из расширения, а из приложение — опять ошибка. Главный смысл, в том, чтобы можно было создать несколько базовых расширений (блог, новости, еще что) — и подключать в нужный момент. (да, да, в django все уже есть, но разговор не о нем) Если брать что-то из расширения — тогда сам смысл расширения пропадает.

Answer 2

[defuz]

Что вам мешает принимать db.session как аргумент вашего расширения? И да, подумайте хорошенько, вам нужен именно extention или blueprint, потому что понятие «блог» больше напоминает именно blueprint.

Comments

[defuz]

Вот вам набросок того что я имею ввиду:

class Blog(object):

     def init(self, app, db):
     	self.app = app
     	self.db = db
     	self.init_models()

 	def init_models(self):
 		# классы в python можно объявить внутри функции
 		class Post(self.db.Model):
 			...blah blah blah...
		self.Post = Post

И используем:

app = Flask('foobar', __name__)
db = SQLAlchemy()

blog = Blog(app, db)

post = blog.Post(title='...', article='...')
db.session.save(post)

[SilentSokolov]

Спасибо, очень интересное решение, надо попробовать, но тут же два вопроса: если расширение большое, хорошо ли запихивать все в один класс, те же вьюхи?

И еще, blueprint, я думал над этим, но опять же если создать «модуль», который в последствии устанавливается pip и регистрируется:

    app.register_blueprint(blog, url_prefix='/blog')

То он тоже не видит базы или я ошибаюсь?

[defuz]

1. Никто же не мешает вам разнести модуль на несколько класов, использовать вместо классов обычные функции и т.д. Единственное, что вам с чем вам прийдется смирится — это с объявлением ваших моделей внутри функции/метода, а не в глобальном контексте.

2. app.register_blueprint не достаточно, чтобы получить доступ к db. Но никто же вам не мешает опять же унаследоватся от blueprint или вызывать app.register_bluprint в методе init так же, как я предложил объявлять модели.

Опять же, пример:

class Blog(object):

     def init(self, app, db, url_prefix):
     	self.app = app
     	self.db = db
     	self.init_models()
        self.init_blueprint()

 	def init_routes(self):
             self.blueprint = Blueprint('blog', __name__)
             self.app.register_blueprint(self.blueprint, url_prefix=url_prefix)
             self.blueprint.add_url_rule('/post/<int:post_id>', self.get_post)
             ... объявляем здесь другие роуты ... 

        def get_post(self, post_id):
               return render_template('post.html', post=self.Post.filter(id=post_id).first())

Такой подход дает вам массу плюшек:
1. Вы можете переопределять любую статику/шаблоны в конкретном приложение
2. Вы можете переопределять вьюшки а даже сами модели.
3. Модульное использование функции url_for.

И да, не путайте flask extention и python module. Никто вам не мешает создать модуль который устанавливается через pip и т.п., который в тоже время не является flask extention. С другой стороны, вам никто не запрещает создать flask extention, задача которго как раз и будет в том числе подключить bluprint к конкретному приложению.

Можете изучить исходники готовых проектов, к примеру flask-social.

[defuz]

Немного напутал в коде, вместо init_routes должен быть init_blueprint, и в него нужно передавать url_prefix.