app.py 7.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254
  1. import contextlib
  2. import os
  3. import flask
  4. import sys
  5. import lc.config as c
  6. import lc.error as e
  7. import lc.model as m
  8. import lc.request as r
  9. import lc.view as v
  10. from lc.web import Endpoint, endpoint, render
  11. app = c.app
  12. @endpoint("/")
  13. class Index(Endpoint):
  14. def html(self):
  15. return render(
  16. "main",
  17. v.Page(
  18. title="main",
  19. content=render(
  20. "message",
  21. v.Message(
  22. title="Lament Configuration",
  23. message="Bookmark organizing for real pinheads.",
  24. ),
  25. ),
  26. user=self.user,
  27. ),
  28. )
  29. @endpoint("/auth")
  30. class Auth(Endpoint):
  31. def api_post(self):
  32. u, token = m.User.login(self.request_data(r.User))
  33. flask.session["auth"] = token
  34. return self.api_ok(u.base_url(), {"token": token})
  35. @endpoint("/login")
  36. class Login(Endpoint):
  37. def html(self):
  38. return render(
  39. "main", v.Page(title="login", content=render("login"), user=self.user,)
  40. )
  41. @endpoint("/logout")
  42. class Logout(Endpoint):
  43. def html(self):
  44. if "auth" in flask.session:
  45. del flask.session["auth"]
  46. raise e.LCRedirect("/")
  47. def api_post(self):
  48. if "auth" in flask.session:
  49. del flask.session["auth"]
  50. return self.api_ok("/")
  51. @endpoint("/u")
  52. class CreateUser(Endpoint):
  53. def html(self):
  54. if self.user:
  55. raise e.LCRedirect(f"/u/{self.user.name}")
  56. token = flask.request.args.get("token")
  57. if not token:
  58. raise e.LCRedirect("/")
  59. return render(
  60. "main",
  61. v.Page(title="add user", user=self.user, content=render("add_user"),),
  62. )
  63. def api_post(self):
  64. token = flask.request.args["token"]
  65. req = self.request_data(r.NewUser).to_user_request()
  66. u = m.User.from_invite(req, token)
  67. flask.session["auth"] = req.to_token()
  68. return self.api_ok(u.base_url(), u.to_dict())
  69. @endpoint("/u/<string:slug>")
  70. class GetUser(Endpoint):
  71. def html(self, slug: str):
  72. u = m.User.by_slug(slug)
  73. pg = int(flask.request.args.get("page", 1))
  74. tags = u.get_tags()
  75. links, pages = u.get_links(as_user=self.user, page=pg)
  76. linklist = v.LinkList(links=links, pages=pages, tags=tags)
  77. return render(
  78. "main",
  79. v.Page(
  80. title=f"user {u.name}",
  81. content=render("linklist", linklist),
  82. user=self.user,
  83. ),
  84. )
  85. def api_get(self, slug: str):
  86. return m.User.by_slug(slug).to_dict()
  87. @endpoint("/u/<string:user>/config")
  88. class GetUserConfig(Endpoint):
  89. def html(self, user: str):
  90. u = self.require_authentication(user)
  91. status_msg = flask.request.args.get("m", None)
  92. if status_msg is not None:
  93. status_msg = int(status_msg)
  94. return render(
  95. "main",
  96. v.Page(
  97. title="configuration",
  98. content=render("config", u.get_config(status_msg)),
  99. user=self.user,
  100. ),
  101. )
  102. @endpoint("/u/<string:user>/invite")
  103. class CreateInvite(Endpoint):
  104. def api_post(self, user: str):
  105. u = self.require_authentication(user)
  106. invite = m.UserInvite.manufacture(u)
  107. return self.api_ok(f"/u/{user}/config", {"invite": invite.token})
  108. @endpoint("/u/<string:user>/password")
  109. class ChangePassword(Endpoint):
  110. def api_post(self, user: str):
  111. u = self.require_authentication(user)
  112. config_url = u.config_url()
  113. req = self.request_data(r.PasswordChange)
  114. try:
  115. req.require_match()
  116. except e.MismatchedPassword:
  117. raise e.LCRedirect(f"{config_url}?m=2")
  118. try:
  119. u.change_password(req)
  120. except e.BadPassword:
  121. raise e.LCRedirect(f"{config_url}?m=3")
  122. return self.api_ok(f"{config_url}?m=1")
  123. @endpoint("/u/<string:user>/l")
  124. class CreateLink(Endpoint):
  125. def html(self, user: str):
  126. url = flask.request.args.get("url", "")
  127. name = flask.request.args.get("name", "")
  128. defaults = v.AddLinkDefaults(name=name, url=url,)
  129. return render(
  130. "main",
  131. v.Page(
  132. title="login", content=render("add_link", defaults), user=self.user,
  133. ),
  134. )
  135. def api_post(self, user: str):
  136. u = self.require_authentication(user)
  137. req = self.request_data(r.Link)
  138. l = m.Link.from_request(u, req)
  139. return self.api_ok(l.link_url(), l.to_dict())
  140. @endpoint("/u/<string:user>/l/<string:link>")
  141. class GetLink(Endpoint):
  142. def api_get(self, user: str, link: str):
  143. u = self.require_authentication(user)
  144. l = u.get_link(int(link))
  145. return self.api_ok(l.link_url(), l.to_dict())
  146. def api_post(self, user: str, link: str):
  147. u = self.require_authentication(user)
  148. l = u.get_link(int(link))
  149. req = self.request_data(r.Link)
  150. l.update_from_request(u, req)
  151. raise e.LCRedirect(l.link_url())
  152. def api_delete(self, user: str, link: str):
  153. u = self.require_authentication(user)
  154. u.get_link(int(link)).delete_instance()
  155. return self.api_ok(u.base_url())
  156. def html(self, user: str, link: str):
  157. l = m.User.by_slug(user).get_link(int(link))
  158. return render(
  159. "main",
  160. v.Page(
  161. title=f"link {l.name}",
  162. content=render("linklist", v.LinkList([l.to_view(self.user)], [])),
  163. user=self.user,
  164. ),
  165. )
  166. @endpoint("/u/<string:slug>/l/<string:link>/edit")
  167. class EditLink(Endpoint):
  168. def html(self, slug: str, link: str):
  169. u = self.require_authentication(slug)
  170. l = u.get_link(int(link))
  171. return render(
  172. "main",
  173. v.Page(
  174. title="login",
  175. content=render("edit_link", v.SingleLink(l)),
  176. user=self.user,
  177. ),
  178. )
  179. @endpoint("/u/<string:user>/t/<path:tag>")
  180. class GetTaggedLinks(Endpoint):
  181. def html(self, user: str, tag: str):
  182. u = m.User.by_slug(user)
  183. pg = int(flask.request.args.get("page", 0))
  184. t = u.get_tag(tag)
  185. links, pages = t.get_links(as_user=self.user, page=pg)
  186. tags = u.get_related_tags(t)
  187. linklist = v.LinkList(links=links, pages=pages, tags=tags)
  188. return render(
  189. "main",
  190. v.Page(
  191. title=f"tag {tag}",
  192. content=render("linklist", linklist),
  193. user=self.user,
  194. ),
  195. )
  196. @endpoint("/u/<string:user>/import")
  197. class PinboardImport(Endpoint):
  198. def html(self, user: str):
  199. u = self.require_authentication(user)
  200. return render(
  201. "main",
  202. v.Page(
  203. title=f"import pinboard data", content=render("import"), user=self.user,
  204. ),
  205. )
  206. def api_post(self, user: str):
  207. u = self.require_authentication(user)
  208. if "file" not in flask.request.files:
  209. raise e.BadFileUpload("could not find attached file")
  210. file = flask.request.files["file"]
  211. if file.filename == "":
  212. raise e.BadFileUpload("no file selected")
  213. u.import_pinboard_data(file.stream)
  214. return self.api_ok(u.base_url())