Update: Fixed bug

platforms/pentesterland.py

@@ -34,24 +34,40 @@ def get_response(self) -> list:
 
  # Extract data from pentester.land
  for row in find_table.findAll("tr"):
+
+ # Check if author has link
  try:
  _author_row = row.findAll("a")[1]
  _post_row = row.findAll("a")[0]
+ except IndexError as e:
+ get_row = row.findAll("td")
+ try:
+ _author_row = get_row[1]
+ _post_row = get_row[0].findAll("a")[0]
+ except IndexError:
+ continue
+
+ # Put Author link None author doesn't have link
+ try:
  _author = dict(
  name=_author_row.text,
  username=_author_row.attrs["href"]
  )
- _post = dict(
- title=_post_row.text,
- link=_post_row.attrs["href"]
- )
- self.output_data.append(
- {
- "author": _author,
- "post": _post
- }
+ except KeyError:
+ _author = dict(
+ name=_author_row.text,
+ username=None
  )
- except IndexError:
- pass
+
+ _post = dict(
+ title=_post_row.text,
+ link=_post_row.attrs["href"]
+ )
+ self.output_data.append(
+ {
+ "author": _author,
+ "post": _post
+ }
+ )
 
  return self.output_data

utils.py

@@ -65,9 +65,17 @@ def notify_writeup(notify_webhook: str, data) -> bool:
  """
  author = data.get("author")
  _post = data.get("post")
- parse_data = f"🟢 New WriteUp !\n\n" \
- f"🖇 WriteUp Link: ||[{_post['title']}]({_post['link']})||\n\n" \
- f"🧑‍💻 Writeup Author: ||[{author['name']}]({author['username']})||\n"
+
+ # Check Author username is None or not
+ if author["username"] is None:
+ parse_data = f"🟢 New WriteUp !\n\n" \
+ f"🖇 WriteUp Link: ||[{_post['title']}]({_post['link']})||\n\n" \
+ f"🧑‍💻 Writeup Author: ||{author['name']}||\n"
+ else:
+ parse_data = f"🟢 New WriteUp !\n\n" \
+ f"🖇 WriteUp Link: ||[{_post['title']}]({_post['link']})||\n\n" \
+ f"🧑‍💻 Writeup Author: ||[{author['name']}]({author['username']})||\n"
+
  data = dict(
  content=parse_data
  )