demodeleter.py

import re
import os
from collections import OrderedDict
#configuration

KillStreakFile = "/home/barnex/test/KillStreaks.txt"
DemoDirectory = "/home/barnex/test"

KillStreakRegex = "\"(.*)\""
DemoFileExtension = ".dem"

demos_with_killstreaks = []

f = open(KillStreakFile, 'r')

for line in f:
  m = re.search(KillStreakRegex, line)
  if m:
    demos_with_killstreaks.append(m.group(1).lower() + DemoFileExtension.lower())

f.close()
print "keeping %d demos" % len(OrderedDict.fromkeys(demos_with_killstreaks).keys())

files_to_delete = [file for file in os.listdir(DemoDirectory) if (file.lower().endswith(DemoFileExtension) and not file.lower() in demos_with_killstreaks)]

#print "DEBUG"
#print "-> demos kept:"
#for demo in demos_with_killstreaks:
  #print "  -> %s" % demo

#print "-> demos deleted:"
#for demo in files_to_delete:
  #print "  -> %s" % demo

for file in files_to_delete:
  #print "deleting: %s" % file
  os.remove(DemoDirectory + "/" + file)

print "deleted %d demos" % len(OrderedDict.fromkeys(files_to_delete).keys())