Example #1
0
    def post(self):
        from csrffilter import CSRFFilter

        user = users.get_current_user()
        mode = self.request.get('mode')

        #CSRF Protection
        filter = CSRFFilter(self, user)
        if not filter.checkCSRFToken():
            return filter.redirectCSRFWarning()

        user_prefs_query = UserPrefs.all()
        user_prefs_query.filter("google_id =", user)
        user_prefs = user_prefs_query.get()

        if user_prefs is not None:
            logging.info('Mode: %s' % mode)
            if mode == 'add':
                keyword = self.request.get('keyword')
                search_keywords = SearchKeywords()
                search_keywords.user_id = user_prefs.key()
                search_keywords.keyword = keyword
                search_keywords.put()

            elif mode == 'delete':
                deletes = self.request.get_all('deletes[]')
                for delete in deletes:
                    search_keywords = SearchKeywords.get_by_id(int(delete))
                    logging.info('Delete keyword: %s' %
                                 search_keywords.keyword)
                    if search_keywords is not None:
                        if search_keywords.user_id.key() == user_prefs.key():
                            search_keywords.delete()

        self.redirect('/config')
Example #2
0
 def get(self):
     search_keywords_query = SearchKeywords().all()
     search_keywords = search_keywords_query.fetch(30)
     for search_keyword in search_keywords:
         try:
             logging.info('Add taskqueue. keyword: %s' %
                          search_keyword.keyword)
             taskqueue.add(url='/feed/keyword',
                           method='GET',
                           params={'keyword_id': search_keyword.key().id()})
         except (taskqueue.Error, apiproxy_errors.Error):
             logging.exception('Failed to add taskqueue.')