def post():
opml_file = request.files['opml_file']
try:
subscriptions = opml.from_string(opml_file.read())
except Exception as error:
raise UnprocessableEntity("Couldn't parse OPML file (%r)" % error)
ccontr = CategoryController(current_identity.id)
fcontr = FeedController(current_identity.id)
counts = {'created': 0, 'existing': 0, 'failed': 0, 'exceptions': []}
categories = {cat.name: cat.id for cat in ccontr.read()}
for line in subscriptions:
try:
link = line.xmlUrl
except Exception as error:
counts['failed'] += 1
counts['exceptions'].append(str(error))
continue
# don't import twice
if fcontr.read(link=link).count():
counts['existing'] += 1
continue
# handling categories
cat_id = None
category = getattr(line, 'category', '').lstrip('/')
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))
counts['created'] += 1
code = 200
if counts.get('created'):
code = 201
elif counts.get('failed'):
code = 400
return counts, code
def populate_db():
fcontr = FeedController()
ccontr = CategoryController()
UserController().create(
**{
'is_admin': True,
'is_api': True,
'cluster_enabled': False,
'login': '******',
'password': '******'
})
user1, user2 = [
UserController().create(login=name,
cluster_enabled=False,
email="*****@*****.**" % name,
password=name) for name in ["user1", "user2"]
]
for iteration in range(2):
article_total = 0
for user in (user1, user2):
for iter_cat in range(3):
cat_id = None
if iter_cat:
cat_id = ccontr.create(user_id=user.id,
name=to_name(
user, iteration, iter_cat)).id
feed_id = fcontr.create(
link="feed%d%d" % (iteration, iter_cat),
user_id=user.id,
category_id=cat_id,
title=to_name(user, iteration, iter_cat, iter_cat)).id
for iter_art in range(3):
entry = to_name(user, iteration, iter_cat, iter_cat,
iter_art)
tags = [
to_name(user, iteration, iter_cat, iter_cat, iter_art,
str(i)) for i in range(2)
]
article_total += 1
ArticleController().create(
entry_id=entry,
link='http://test.te/%d' % article_total,
feed_id=feed_id,
user_id=user.id,
tags=tags,
category_id=cat_id,
title=entry,
date=utc_now() + timedelta(seconds=iteration),
content="content %d" % article_total)
session.commit()
session.flush()
ClusterController().clusterize_pending_articles()
def test_cluster_disabled_on_original_category(self):
article = ArticleController().read(category_id__ne=None).first()
art_cat_id = article.category_id
cat_ctrl = CategoryController(article.user_id)
cluster = article.cluster
fctrl = FeedController(cluster.user_id)
feed = fctrl.create(title='new feed', category_id=art_cat_id)
fno_cat = fctrl.create(title='category-less')
update_on_all_objs(users=[cluster.user], cluster_enabled=None)
cat_ctrl.update({}, {'cluster_enabled': False})
article = self.create_article_from(cluster, feed)
self.assertEqual(1, len(article.cluster.articles))
self.assertNotInCluster(article, cluster)
article = self.create_article_from(cluster, fno_cat)
self.assertEqual(1, len(article.cluster.articles))
self.assertNotInCluster(article, cluster)
cat_ctrl.update({'id': art_cat_id}, {'cluster_enabled': True})
article = self.create_article_from(cluster, fno_cat)
self.assertEqual(2, len(article.cluster.articles))
self.assertInCluster(article, cluster)
article = self.create_article_from(cluster, feed)
self.assertEqual(3, len(article.cluster.articles))
self.assertInCluster(article, cluster)