def get(self, id): '''This method receives an ID from an page and returns the page''' if Commons.isValidId(id): page = Page.objects(id=id) if Commons.checkIfNotExists(page): return Commons.notFound('page') if auth.isAuthorized(page[0].userId): Page.objects(id=id).update(views=page[0].views + 1) return make_response(jsonify({'data': page}), 201) return auth.unauthorized()
def put(self, id): '''This method receives an ID from an page and updates the page''' params = self.reqparse.parse_args() if Commons.isValidId(id): page = Page.objects(id=id) if Commons.checkIfNotExists(page): return Commons.notFound('page') if auth.isAuthorized(page[0].userId): data = Commons.filterQueryParams(params) Page.objects(id=id).update_one(upsert=False, write_concern=None, **data) return make_response(jsonify({'data': 'Page updated'}), 201) return auth.unauthorized()
def get(self, id): if Commons.isValidId(id): pages = Page.objects(categoryId=id) if Commons.checkIfNotExists(pages): return Commons.notFound('page') if auth.isAuthorized(pages[0].userId): return make_response(jsonify({'data': pages}), 201) return auth.unauthorized()
def delete(self, id): '''This method receives an ID from an page and deletes the page''' if Commons.isValidId(id): page = Page.objects(id=id) if Commons.checkIfNotExists(page): return Commons.notFound('page') if auth.isAuthorized(page[0].userId): page.delete() return make_response(jsonify({'data': 'Page was deleted'}), 201) return auth.unauthorized()
def get_state(self): url = self.driver.current_url page = Page.objects(url=url).first() if page is None: self.driver.get(url) default_state = state_builder.get_current_state(self.driver) default_state.name = self.page default_state.save() page = Page(url=url, default_state=default_state, states=[default_state]) page.name = self.page page.save() for state in page.states: if state.name == self.page: print "Found state %s" % state.name return state print "State not found, creating new state" new_state = state_builder.get_current_state(self.driver) new_state.save() new_state.name = self.page page.states.append(new_state) page.save() return new_state
def get(self): '''This method returns all pages from an user''' pages = Page.objects.all() if auth.isAdmin() else Page.objects( userId=auth.user['id']) return Commons.notFound('page') if Commons.checkIfNotExists( pages) else make_response(jsonify({'data': pages}), 201)