def init():
try:
path = os.path.dirname(consts.KEYWORDJOURNAL_DB_PATH)
os.mkdir(path)
except OSError:
pass
Base.metadata.create_all(bind=engine)
db.add(user1)
db.add(user2)
db.commit()
def create(user_id, post_title, post_body):
posted_keywords = parser.find_all_keywords_with_args(post_body)
user_keywords_rs = user_keyword_resource.get_all_keywords_with_args(user_id)
posted_keywords_rs = _construct_posted_keywords(posted_keywords, user_keywords_rs)
post = Post(
user_id=user_id,
text=post_body,
header=post_title,
datetime=datetime.datetime.now(),
posted_keywords=posted_keywords_rs,
)
db.add(post)
db.commit()
def create(user_id, post_title, post_body):
posted_keywords = parser.find_all_keywords_with_args(post_body)
user_keywords_rs = user_keyword_resource.get_all_keywords_with_args(
user_id)
posted_keywords_rs = _construct_posted_keywords(posted_keywords,
user_keywords_rs)
post = Post(
user_id=user_id,
text=post_body,
header=post_title,
datetime=datetime.datetime.now(),
posted_keywords=posted_keywords_rs,
)
db.add(post)
db.commit()
return post
def create(keyword, user_id):
keyword_res = resources.keyword.get(keyword)
if keyword_res is None:
# Possible race condition
try:
keyword_res = resources.keyword.create(keyword)
except IntegrityError:
db.rollback()
keyword_res = resources.keyword.get(keyword)
user_keyword = UserKeyword(
keyword_id=keyword_res.id,
user_id=user_id,
)
try:
db.add(user_keyword)
db.commit()
return user_keyword
except IntegrityError:
db.rollback()
return None
# Author: Daniel Kinney
# All rights reserved
from keywordjournal.kwj_flask.connection.manager import db, engine
from keywordjournal.models.db import *
keyword1 = Keyword(word="kw1")
keyword2 = Keyword(word="kw2")
user1 = User(
email="*****@*****.**",
password="******",
user_keywords=[UserKeyword(keyword=keyword1, user_args=[UserArg(param_name="x"), UserArg(param_name="y")])],
)
user2 = User(email="*****@*****.**", password="******")
if __name__ == "__main__":
Base.metadata.create_all(bind=engine)
db.add(user1)
db.add(user2)
db.commit()
def create(word):
new_keyword = Keyword(word=word)
db.add(new_keyword)
db.commit()
return new_keyword