Python DBSession.add方法代码示例

# 需要导入模块: from tagger.model import DBSession [as 别名]
# 或者: from tagger.model.DBSession import add [as 别名]
    def test_banner_set(self):
        """controllers.admin.Controller.banner_set is working properly"""
        languageid, tadm = self._fill_db()
        media = Media(u'image', u'test image', u'/test.png', tadm, u'xx',
                                                                u'random text')
        DBSession.add(media)
        link = Link(u'test link', u'http://example.com', tadm, u'xx',
                                                                u'random text')
        DBSession.add(link)
        DBSession.flush()
        mediaid = media.id
        linkid = link.id
        transaction.commit()
        
        environ = {'REMOTE_USER': 'test_admin'}
        response = self.app.post('/admin/banner_set',
                                            dict(mediaid=mediaid,
                                                 linkid=linkid,
                                                ),
                                            extra_environ=environ, status=302)
        #redirected = response.follow(extra_environ=environ, status=200)

        bc = DBSession.query(BannerContent).first()
        eq_(bc.media_id, mediaid)
        eq_(bc.link_id, linkid)

# 需要导入模块: from tagger.model import DBSession [as 别名]
# 或者: from tagger.model.DBSession import add [as 别名]
 def _fill_db(self):
     language = Language(u'xx', u'test language')
     DBSession.add(language)
     DBSession.flush()
     languageid = language.id
     transaction.commit()
     return languageid.encode()

# 需要导入模块: from tagger.model import DBSession [as 别名]
# 或者: from tagger.model.DBSession import add [as 别名]
    def test_comment(self):
        """controllers.admin.Controller.comment is working properly"""
        languageid, tadm = self._fill_db()
        comment = Comment(u'anonymous', u'[email protected]', u'test comment')
        DBSession.add(comment)
        DBSession.flush()
        commentid = comment.id
        transaction.commit()
        
        environ = {'REMOTE_USER': 'test_admin'}
        response = self.app.get('/admin/comment/', extra_environ=environ,
                                                                    status=200)

        tr = response.html.table('tr')[1]
        # Date
        eq_(str(tr('td')[1]), '<td>%s</td>' % commentid)
        eq_(str(tr('td')[2]), '<td></td>')
        eq_(str(tr('td')[3]), '<td>anonymous</td>')
        eq_(str(tr('td')[4]), '<td>[email protected]</td>')
        eq_(str(tr('td')[5]), '<td>test comment</td>')
        # Status
        actions = tr('td')[7]
        eq_(str(actions('a')[0]['class']), 'icon edit overlay')
        eq_(str(actions('a')[1]['class']), 'icon delete overlay')
        eq_(str(actions('a')[2]['class']), 'icon approve')
        eq_(str(actions('a')[3]['class']), 'icon spam')

# 需要导入模块: from tagger.model import DBSession [as 别名]
# 或者: from tagger.model.DBSession import add [as 别名]
 def post(self, name, languageid):
     """create a new Tag"""
     user = tmpl_context.user
     tag = Tag(name, languageid)
     DBSession.add(tag)
     flash('%s %s' % (_('Created Tag:'), tag.id), 'ok')
     return dict(redirect_to=url('/admin/tag/'))

# 需要导入模块: from tagger.model import DBSession [as 别名]
# 或者: from tagger.model.DBSession import add [as 别名]
    def test_article(self):
        """controllers.admin.Controller.article is working properly"""
        languageid, tadm = self._fill_db()
        cat = Category(u'test_category', u'xx')
        DBSession.add(cat)
        article = Article(u'A Test Article!', cat, u'xx', tadm, u'random text')
        DBSession.add(article)
        DBSession.flush()
        categoryid = cat.id
        articleid = article.id
        transaction.commit()

        environ = {'REMOTE_USER': 'test_admin'}
        response = self.app.get('/admin/article/', extra_environ=environ,
                                                                    status=200)

        tr = response.html.table.find('tr', articleid)
        # Date
        eq_(str(tr('td')[1]), '<td>test_category</td>')
        eq_(str(tr('td')[2]), '<td>%s</td>' % articleid)
        eq_(str(tr('td')[3]), '<td>A Test Article!</td>')
        eq_(str(tr('td')[4]), '<td></td>')
        eq_(str(tr('td')[5]), '<td>%s</td>' % languageid)
        # Status
        actions = tr('td')[7]
        eq_(str(actions('a')[0]['class']), 'icon edit')
        eq_(str(actions('a')[1]['class']), 'icon delete overlay')
        eq_(str(actions('a')[2]['class']), 'icon publish')

# 需要导入模块: from tagger.model import DBSession [as 别名]
# 或者: from tagger.model.DBSession import add [as 别名]
 def _fill_db(self):
     tadm = DBSession.query(User).filter_by(user_name=u'test_admin').one()
     language = Language(u'xx', u'test language')
     DBSession.add(language)
     DBSession.flush()
     languageid = language.id
     return languageid, tadm

# 需要导入模块: from tagger.model import DBSession [as 别名]
# 或者: from tagger.model.DBSession import add [as 别名]
 def do_get_dependencies(self):
     try:
         self.language = content.Language(id=u'xx', name=u'test_lang')
         DBSession.add(self.language)
         DBSession.flush()
         return dict(lang=self.language.id,
                    )
     except:
         DBSession.rollback()
         raise

# 需要导入模块: from tagger.model import DBSession [as 别名]
# 或者: from tagger.model.DBSession import add [as 别名]
 def _fill_db(self):
     language = Language(u'xx', u'test_langugage')
     DBSession.add(language)
     cat = Category(u'test category', u'xx', u'a test category')
     DBSession.add(cat)
     DBSession.flush()
     languageid = language.id
     categoryid = cat.id
     transaction.commit()
     return languageid, categoryid.encode()

# 需要导入模块: from tagger.model import DBSession [as 别名]
# 或者: from tagger.model.DBSession import add [as 别名]
 def _fill_db(self):
     language = Language(u'xx', u'test_langugage')
     DBSession.add(language)
     tag = Tag(u'test tag', u'xx')
     DBSession.add(tag)
     DBSession.flush()
     languageid = language.id
     tagid = tag.id
     transaction.commit()
     return languageid, tagid.encode()

# 需要导入模块: from tagger.model import DBSession [as 别名]
# 或者: from tagger.model.DBSession import add [as 别名]
    def post(self, name, uri, languageid, description=None, tagids=None):
        """create a new Link"""
        user = tmpl_context.user
        lang = tmpl_context.lang or DBSession.query(Language).first().id
        link = Link(name, uri, user, languageid, description)
        DBSession.add(link)

        tags = tags_from_string(tagids, lang=lang)
        link.tags[:] = tags

        flash('%s %s' % (_('Created Link:'), link.id), 'ok')
        return dict(redirect_to=url('/admin/link/'))

# 需要导入模块: from tagger.model import DBSession [as 别名]
# 或者: from tagger.model.DBSession import add [as 别名]
    def post(self, title, categoryid, languageid, text=None, tagids=None):
        """create a new Article"""
        user = tmpl_context.user
        lang = tmpl_context.lang or DBSession.query(Language).first().id
        category = DBSession.query(Category).get(categoryid)
        article = Article(title, category, languageid, user, text)
        DBSession.add(article)

        tags = tags_from_string(tagids, lang=lang)
        article.tags[:] = tags

        flash('%s %s' % (_('Created Article:'), article.id), 'ok')
        return dict(redirect_to=url('/admin/article/'))

# 需要导入模块: from tagger.model import DBSession [as 别名]
# 或者: from tagger.model.DBSession import add [as 别名]
def tags_from_string(s, create=True, lang=None):
    if lang is None:
        lang = DBSession.query(Language).first().id
    alltags = dict([(t.id, t) for t in DBSession.query(Tag)])
    alltagsdata = dict([(d.name, d) for d in DBSession.query(TagData)])
    tagnames = s and s.split(', ') or []
    tags = []
    for tagname in tagnames:
        if tagname in alltags:
            tag = alltags[tagname]
        elif tagname in alltagsdata:
            tag = alltagsdata[tagname].parent
        elif create:
            tag = Tag(tagname, lang)
            DBSession.add(tag)
        tags.append(tag)
    return tags

# 需要导入模块: from tagger.model import DBSession [as 别名]
# 或者: from tagger.model.DBSession import add [as 别名]
    def test_tag(self):
        """controllers.admin.Controller.tag is working properly"""
        languageid, tadm = self._fill_db()
        tag = Tag(u'test tag', u'xx')
        DBSession.add(tag)
        DBSession.flush()
        tagid = tag.id
        transaction.commit()
        
        environ = {'REMOTE_USER': 'test_admin'}
        response = self.app.get('/admin/tag/', extra_environ=environ,
                                                                    status=200)

        tr = response.html.table('tr')[1]
        eq_(str(tr('td')[0]), '<td>%s</td>' % tagid)
        eq_(str(tr('td')[1]), '<td>test tag</td>')
        eq_(str(tr('td')[2]), '<td>%s</td>' % languageid)
        actions = tr('td')[3]
        eq_(str(actions('a')[0]['class']), 'icon edit overlay')
        eq_(str(actions('a')[1]['class']), 'icon delete overlay')

# 需要导入模块: from tagger.model import DBSession [as 别名]
# 或者: from tagger.model.DBSession import add [as 别名]
    def test_category(self):
        """controllers.admin.Controller.category is working properly"""
        languageid, tadm = self._fill_db()
        category = Category(u'test category', u'xx', u'a test category')
        DBSession.add(category)
        DBSession.flush()
        categoryid = category.id
        transaction.commit()

        environ = {'REMOTE_USER': 'test_admin'}
        response = self.app.get('/admin/category/', extra_environ=environ,
                                                                    status=200)
        
        tr = response.html.table.find('tr', categoryid)
        eq_(str(tr('td')[0]), '<td>%s</td>' % categoryid)
        eq_(str(tr('td')[1]), '<td>test category</td>')
        eq_(str(tr('td')[2]), '<td>a test category</td>')
        eq_(str(tr('td')[3]), '<td>%s</td>' % languageid)
        actions = tr('td')[4]
        eq_(str(actions('a')[0]['class']), 'icon edit overlay')
        eq_(str(actions('a')[1]['class']), 'icon delete overlay')

# 需要导入模块: from tagger.model import DBSession [as 别名]
# 或者: from tagger.model.DBSession import add [as 别名]
    def settings_set(self, name=None, value=None):
        """Set settings values"""
        query = DBSession.query(Setting)
        settings = dict([(s.id, s) for s in query])
        if not len(name) == len(value):
            raise HTTPBadRequest("names and values don't match")

        modified = False
        for n, v in zip(name, value):
            if n in settings:
                if not settings[n].value == v:
                    settings[n].value = v
                    modified = True
            else:
                DBSession.add(Setting(n, v))
                modified = True

        if modified:
            flash(_('Updated Settings'), 'ok')
        else:
            flash(_('Settings are unchanged'), 'info')
        redirect(url('/admin/settings/'))