xiumingzhang
diff --git a/‎run.py‎
Lines changed: 6 additions & 3 deletions b/‎run.py‎
Lines changed: 6 additions & 3 deletions
diff --git a/‎util.py‎
Lines changed: 1 addition & 12 deletions b/‎util.py‎
Lines changed: 1 addition & 12 deletions
@@ -49,9 +49,7 @@ def main(args):
 
  for url, opt in tqdm(roster.items(), desc="Checking URLs"):
  out_dir = join(
- args.tmp_dir,
- url.replace('/', '_').replace('?', '_').replace('&', '_'),
- ).rstrip('/')
+ args.tmp_dir, replace_special_char(url)).rstrip('/')
 
  # Take screenshots
  screenshot(url, out_dir, opt)
@@ -132,5 +130,10 @@ def load_roster(roster_json):
  return roster
 
 
+def replace_special_char(url):
+ return url.replace(
+ '/', '_').replace('?', '_').replace('&', '_').replace(':', '_')
+
+
 if __name__ == '__main__':
  main(parser.parse_args())
@@ -43,18 +43,7 @@ def email_myself(
  to_emails = [email]
  from_email = email
 
- msg="""\
-<html>
- <head></head>
- <body>
- <p>Hi!<br>
- How are you?<br>
- Here is the <a href="https://www.python.org">link</a> you wanted.
- </p>
- </body>
-</html>
-"""
- msg = MIMEText(msg,'html')
+ msg = MIMEText(msg)
  msg['Subject'] = subject
  msg['To'] = ', '.join(to_emails)
  msg['From'] = from_email