Pages also stored in the DB.

talkpython · May 31, 2020 · f0178c0 · f0178c0
1 parent 77c88ac
commit f0178c0
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 45 deletions.
diff --git a/code/flask/ch07-saving-content/pypi_flask_ch07/pypi_org/db/fake_data.py b/code/flask/ch07-saving-content/pypi_flask_ch07/pypi_org/db/fake_data.py
diff --git a/code/flask/ch07-saving-content/pypi_flask_ch07/pypi_org/db/pypi.sqlite b/code/flask/ch07-saving-content/pypi_flask_ch07/pypi_org/db/pypi.sqlite
diff --git a/...flask/ch07-saving-content/pypi_flask_ch07/pypi_org/viewmodels/admin/editpage_viewmodel.py b/...flask/ch07-saving-content/pypi_flask_ch07/pypi_org/viewmodels/admin/editpage_viewmodel.py
@@ -16,9 +16,9 @@ def __init__(self, page_id: int = 0):
  self.page = cms_service.get_page_by_id(self.page_id)
 
  if self.page:
- self.title = self.page.get('title')
- self.url = self.page.get('url')
- self.contents = self.page.get('contents')
+ self.title = self.page.title
+ self.url = self.page.url
+ self.contents = self.page.contents
 
  def process_form(self):
  d = self.request_dict

diff --git a/code/flask/ch07-saving-content/pypi_flask_ch07/pypi_org/viewmodels/cms/request_viewmodel.py b/code/flask/ch07-saving-content/pypi_flask_ch07/pypi_org/viewmodels/cms/request_viewmodel.py
@@ -12,7 +12,7 @@ def __init__(self, url):
  self.page = cms_service.get_page(self.url)
  self.html = None
  if self.page:
- self.html = self.page.get('contents')
+ self.html = self.page.contents
 
  self.redirect = cms_service.get_redirect(self.url)
  self.redirect_url = None

diff --git a/code/flask/ch07-saving-content/pypi_flask_ch07/pypi_org/views/admin_views.py b/code/flask/ch07-saving-content/pypi_flask_ch07/pypi_org/views/admin_views.py
@@ -112,7 +112,7 @@ def add_page_post():
  if not vm.validate():
  return vm.to_dict()
 
- cms_service.create_page(vm.title, vm.url, vm.contents)
+ cms_service.create_page(vm.title, vm.url, vm.contents, vm.user.email)
 
  return flask.redirect('/admin/pages')