def process_form(category_id=None):
form = CategoryForm()
cat_contr = CategoryController(current_user.id)
if not form.validate():
return render_template('edit_category.html', form=form)
existing_cats = list(cat_contr.read(name=form.name.data))
if existing_cats and category_id is None:
flash(gettext("Couldn't add category: already exists."), "warning")
return redirect(
url_for('category.form', category_id=existing_cats[0].id))
# Edit an existing category
category_attr = {'name': form.name.data}
if category_id is not None:
cat_contr.update({'id': category_id}, category_attr)
flash(
gettext('Category %(cat_name)r successfully updated.',
cat_name=category_attr['name']), 'success')
return redirect(url_for('category.form', category_id=category_id))
# Create a new category
new_category = cat_contr.create(**category_attr)
flash(
gettext('Category %(category_name)r successfully created.',
category_name=new_category.name), 'success')
return redirect(url_for('category.form', category_id=new_category.id))
def process_form(category_id=None):
form = CategoryForm()
cat_contr = CategoryController(g.user.id)
if not form.validate():
return render_template('edit_category.html', form=form)
existing_cats = list(cat_contr.read(name=form.name.data))
if existing_cats and category_id is None:
flash(gettext("Couldn't add category: already exists."), "warning")
return redirect(url_for('category.form',
category_id=existing_cats[0].id))
# Edit an existing category
category_attr = {'name': form.name.data}
if category_id is not None:
cat_contr.update({'id': category_id}, category_attr)
flash(gettext('Category %(cat_name)r successfully updated.',
cat_name=category_attr['name']), 'success')
return redirect(url_for('category.form', category_id=category_id))
# Create a new category
new_category = cat_contr.create(**category_attr)
flash(gettext('Category %(category_name)r successfully created.',
category_name=new_category.name), 'success')
return redirect(url_for('category.form', category_id=new_category.id))
def populate_db():
db_create()
ucontr = UserController()
ccontr = CategoryController()
fcontr = FeedController()
acontr = ArticleController()
ccontr = CategoryController()
user1, user2 = [
ucontr.create(login=name, email="*****@*****.**" % name, password=name)
for name in ["user1", "user2"]
]
article_total = 0
for user in (user1, user2):
for i in range(3):
cat_id = None
if i:
cat_id = ccontr.create(user_id=user.id,
name="category%d" % i).id
feed = fcontr.create(link="feed%d" % i,
user_id=user.id,
category_id=cat_id,
title="%s feed%d" % (user.login, i))
for j in range(3):
entry = "%s %s article%d" % (user.login, feed.title, j)
article_total += 1
acontr.create(entry_id=entry,
link='http://test.te/%d' % article_total,
feed_id=feed.id,
user_id=user.id,
category_id=cat_id,
title=entry,
content="content %d" % article_total)
def populate_db():
db_create()
ucontr = UserController()
ccontr = CategoryController()
fcontr = FeedController()
acontr = ArticleController()
ccontr = CategoryController()
user1, user2 = [ucontr.create(login=name, email="*****@*****.**" % name,
password=name)
for name in ["user1", "user2"]]
article_total = 0
for user in (user1, user2):
for i in range(3):
cat_id = None
if i:
cat_id = ccontr.create(user_id=user.id,
name="category%d" % i).id
feed = fcontr.create(link="feed%d" % i, user_id=user.id,
category_id=cat_id,
title="%s feed%d" % (user.login, i))
for j in range(3):
entry = "%s %s article%d" % (user.login, feed.title, j)
article_total += 1
acontr.create(entry_id=entry,
link='http://test.te/%d' % article_total,
feed_id=feed.id, user_id=user.id, category_id=cat_id,
title=entry, content="content %d" % article_total)
def opml_import():
if request.files.get('opmlfile', None) is None:
flash(gettext('Got no file'), 'warning')
return redirect(url_for('user.profile'))
data = request.files.get('opmlfile', None)
try:
subscriptions = opml.from_string(data.read())
except:
flash(gettext("Couldn't parse file"), 'danger')
return redirect(request.referrer)
ccontr = CategoryController(current_user.id)
fcontr = FeedController(current_user.id)
created_count, existing_count, failed_count = 0, 0, 0
categories = {cat.name: cat.id for cat in ccontr.read()}
for line in subscriptions:
try:
link = line.xmlUrl
except Exception:
failed_count += 1
continue
# don't import twice
if fcontr.read(link=link).count():
existing_count += 1
continue
# handling categories
cat_id = None
category = getattr(line, 'category', None)
if category:
if category not in categories:
new_category = ccontr.create(name=category)
categories[new_category.name] = new_category.id
cat_id = categories[category]
fcontr.create(title=getattr(line, 'text', None),
category_id=cat_id,
description=getattr(line, 'description', None),
link=link,
site_link=getattr(line, 'htmlUrl', None))
created_count += 1
flash(
gettext(
"Created %(created)d feed ! (%(failed)d import failed, "
"%(existing)d were already existing)",
created=created_count,
failed=failed_count,
existing=existing_count), "info")
return redirect(url_for('user.profile'))
def opml_import():
if request.files.get('opmlfile', None) is None:
flash(gettext('Got no file'), 'warning')
return redirect(url_for('user.profile'))
data = request.files.get('opmlfile', None)
try:
subscriptions = opml.from_string(data.read())
except:
flash(gettext("Couldn't parse file"), 'danger')
return redirect(request.referrer)
ccontr = CategoryController(current_user.id)
fcontr = FeedController(current_user.id)
created_count, existing_count, failed_count = 0, 0, 0
categories = {cat.name: cat.id for cat in ccontr.read()}
for line in subscriptions:
try:
link = line.xmlUrl
except Exception:
failed_count += 1
continue
# don't import twice
if fcontr.read(link=link).count():
existing_count += 1
continue
# handling categories
cat_id = None
category = getattr(line, 'category', None)
if category:
if category not in categories:
new_category = ccontr.create(name=category)
categories[new_category.name] = new_category.id
cat_id = categories[category]
fcontr.create(title=getattr(line, 'text', None), category_id=cat_id,
description=getattr(line, 'description', None),
link=link, site_link=getattr(line, 'htmlUrl', None))
created_count += 1
flash(gettext("Created %(created)d feed ! (%(failed)d import failed, "
"%(existing)d were already existing)",
created=created_count, failed=failed_count,
existing=existing_count), "info")
return redirect(url_for('user.profile'))
