Main API Development (#13)

* feat: config, jinja2 and requests, baseclass

* chore: move class modules to classes subpackage

* fix: move config.py to classes subpackage

* fix(flake8): change max line length to 88

* fix: change _config to property

* feat: add working user and post classes + api changes

* chore: add newline at end of .flake8 file

Authored: BD103 <dont@stalk.me>

Author: BD103

.flake8

@@ -1,2 +1,3 @@
 [flake8]
-indent-size = 2
+indent-size = 2
+max-line-length = 88

poetry.lock

@@ -9,6 +9,8 @@ repository = "https://github.com/ReplAPI-it/Python-ReplAPI-It"
 
 [tool.poetry.dependencies]
 python = "^3.8"
+jinja2 = "^3.0.1"
+requests = "^2.25.1"
 
 [tool.poetry.dev-dependencies]
 isort = "^5.8.0"

pyproject.toml

@@ -0,0 +1,4 @@
+from .base import BaseClass  # noqa: F401
+from .config import config  # noqa: F401
+from .post import Post  # noqa: F401
+from .user import User  # noqa: F401

replapi_it/classes/__init__.py

@@ -0,0 +1,51 @@
+import json
+
+import requests
+from jinja2 import Template
+
+from .config import config
+
+_query_template = """
+query Main({% for i in query_args %}{{ i }}: {{ query_args[i] }}, {% endfor %}) {
+  {% for i in types %}
+  {{ i }} {
+    {% for field in types[i] %}
+    {{ field }}
+    {% endfor %}
+  }
+  {% endfor %}
+}
+""".strip()
+
+_default_headers = {
+  "X-Requested-With": "ReplAPI-It-Python",
+  "referrer": config["graphql_url"],
+}
+
+
+class BaseClass(object):
+  def __init__(self, vars: dict = {}, **kwargs):
+    self.vars = {**vars, **kwargs}
+    self._query_args = {"$username": "String!"}
+    self._types = {"userByUsername(username: $username)": ["fullName", "karma"]}
+
+  @property
+  def config(self) -> dict:
+    return config
+
+  def query(self) -> str:
+    tm = Template(_query_template)
+
+    return tm.render(query_args=self._query_args, types=self._types)
+
+  def collect_json(self):
+    req = requests.post(
+      self.config["graphql_url"],
+      data={"query": self.query(), "variables": json.dumps(self.vars)},
+      headers=_default_headers,
+    )
+
+    return req.text
+
+  def collect(self):
+    return json.loads(self.collect_json())

replapi_it/classes/base.py

@@ -0,0 +1,15 @@
+import json
+import os
+
+default = {"graphql_url": "https://staging.replit.com/graphql"}
+
+
+def _config(filename: str = "replapi_it.json") -> dict:
+  if os.path.isfile(filename):
+    with open(filename, "rt") as fp:
+      return json.load(fp)
+  else:
+    return {}
+
+
+config = {**default, **_config()}

replapi_it/classes/config.py

@@ -0,0 +1,36 @@
+from .base import BaseClass
+
+
+class Post(BaseClass):
+  def __init__(self, id: int, **kwargs):
+    self.vars = {**{"id": id}, **kwargs}
+    self._query_args = {"$id": "Int!"}
+    self._types = {
+      "post(id: $id)": [
+        "id",
+        "title",
+        "body",
+        "showHosted",
+        "voteCount",
+        "commentCount",
+        "isPinned",
+        "isLocked",
+        "timeCreated",
+        "timeUpdated",
+        "url",
+        "isAnnouncement",
+        "isAuthor",
+        "canEdit",
+        "canComment",
+        "canVote",
+        "canPin",
+        "canSetType",
+        "canChangeBoard",
+        "canLock",
+        "hasVoted",
+        "canReport",
+        "hasReported",
+        "isAnswerable",
+        "tutorialPages",
+      ]
+    }

replapi_it/classes/post.py

@@ -0,0 +1,27 @@
+from .base import BaseClass
+
+
+class User(BaseClass):
+  def __init__(self, username: str, **kwargs):
+    self.vars = {**{"username": username}, **kwargs}
+    self._query_args = {"$username": "String!"}
+    self._types = {
+      "userByUsername(username: $username)": [
+        "id",
+        "username",
+        "firstName",
+        "lastName",
+        "bio",
+        "isVerified",
+        "displayName",
+        "fullName",
+        "url",
+        "isLoggedIn",
+        "isSubscribed",
+        "timeCreated",
+        "isBannedFromBoards",
+        "karma",
+        "isHacker",
+        "image",
+      ]
+    }

replapi_it/classes/user.py

@@ -1,5 +1,4 @@
-from .post import _Post
-from .user import _User
+from . import classes
 
 defaultInitVars = {"username": None}
 
@@ -12,5 +11,5 @@ class ReplAPI(object):
   def __init__(self, initVars: dict = {}):
     self.vars = {**defaultInitVars, **initVars}
 
-    self.User = _User
-    self.Post = _Post
+    self.User = classes.User
+    self.Post = classes.Post