Jump to content
The Dark Mod Forums

Help with Python Script


Destined

Recommended Posts

Hi all,

some time ago SteveL gave me a python script, that can search files in the TDM installation (also inside pk4-files), which is very handy for playing around with definitions, materials, etc. Unfortunately, the script does not work with the latest Python distribution. Could someone, who in contrast to myself knows python, be so kind to correct the script, so it works? Idle gives an error that the print part is wrong; I am not sure if they have changed anything in their syntax. Any help is appreciated!

"""Search text files in FMs and TDM assets"""
import re, os, fnmatch, zipfile

PATHS = [ r"E:\Spiele\Stealth\TDM" ] #, r"E:\dm-dev\campaign"]  
FILE_EXTS = ['.mtr'] # '.mtr', '.prt', ,'.skin','.fx','.md5mesh','.pfb', '.anim', '.md5anim','.gui','.map','.script','.gui','.def'
SEARCH_PTN = r'arm_leg' # regular expression

def searchPk4(pk4, exts):
    """Look in a pk4 archive and yield files with any of the extensions
    in exts. Returns ( filename, filecontents ). filename includes any
    directory path internal to the archive.
    
    pk4: full path, e.g. r'E:\darkmod\tdm_defs01.pk4'
    exts: list or tuple of bare extensions, e.g. ['def', 'mtr']
    """
    f = zipfile.ZipFile(pk4, 'r')
    for member in f.infolist():
        ext = (os.path.splitext(member.filename)[1]).lower()
        if ext in exts:
            yield member.filename, f.read(member)

def findTDMFiles(paths, exts):
    """Yield tuples: (filepath, filecontents) of all files
    found in paths with any of the extensions in exts. Looks inside
    pk4 archives, but not nested archives.
    
    paths, exts are sequences containing one or more paths / extensions.
    """
    for path in set(paths):
        for root, dirnames, filenames in os.walk(path):
            for ext in set( ['.pk4'] + exts):
                for filename in fnmatch.filter(filenames, '*'+ext):
                    fpath = os.path.join(root, filename)
                    if ext == '.pk4':
                        for internalname, content in searchPk4(fpath, exts):
                            yield ( fpath + ' -> ' + internalname,   content )
                    else:
                         yield fpath, file(fpath).read()

if __name__ == '__main__':
    search_regex = re.compile(SEARCH_PTN, re.IGNORECASE)
    context_regex = re.compile("[^\n]{0,250}(" + SEARCH_PTN + ")[^\n]{0,250}", re.IGNORECASE)

    result = '\n~~~~~ RESULT:\n'
    for fname, text in findTDMFiles(PATHS, FILE_EXTS):
        print "Checking " + fname
        refs = search_regex.findall(text)
        if refs:
            print '***** %s *****' % fname
            result += '***** %s *****\n' % fname
            for r in context_regex.finditer(text):
                print r.group().strip()
                result += r.group().strip() + '\n'
    print result

Link to comment
Share on other sites

In Python 3 you can no longer write:

print "something"

it has to be

print("something")

as if you are calling a function.

 

I have no idea if that's the only problem with the script, but it would be the first thing to try if you are getting errors about print statements.

 

Also be aware that you are not forced to run scripts with Python 3. You should be able to install and run Python 2 quite happily alongside Python 3, and most Linux distributions will do this by default (because there are a lot of existing scripts which have not been updated).

  • Like 2
Link to comment
Share on other sites

Join the conversation

You can post now and register later. If you have an account, sign in now to post with your account.

Guest
Reply to this topic...

×   Pasted as rich text.   Paste as plain text instead

  Only 75 emoji are allowed.

×   Your link has been automatically embedded.   Display as a link instead

×   Your previous content has been restored.   Clear editor

×   You cannot paste images directly. Upload or insert images from URL.

  • Recent Status Updates

    • The Black Arrow

      Hope everyone has the blessing of undying motivation for "The Dark Mod 15th Anniversary Contest". Can't wait to see the many magnificent missions you all may have planned. Good luck, with an Ace!
      · 0 replies
    • Ansome

      Finally got my PC back from the shop after my SSD got corrupted a week ago and damaged my motherboard. Scary stuff, but thank goodness it happened right after two months of FM development instead of wiping all my work before I could release it. New SSD, repaired Motherboard and BIOS, and we're ready to start working on my second FM with some added version control in the cloud just to be safe!
      · 2 replies
    • Petike the Taffer  »  DeTeEff

      I've updated the articles for your FMs and your author category at the wiki. Your newer nickname (DeTeEff) now comes first, and the one in parentheses is your older nickname (Fieldmedic). Just to avoid confusing people who played your FMs years ago and remember your older nickname. I've added a wiki article for your latest FM, Who Watches the Watcher?, as part of my current updating efforts. Unless I overlooked something, you have five different FMs so far.
      · 0 replies
    • Petike the Taffer

      I've finally managed to log in to The Dark Mod Wiki. I'm back in the saddle and before the holidays start in full, I'll be adding a few new FM articles and doing other updates. Written in Stone is already done.
      · 4 replies
    • nbohr1more

      TDM 15th Anniversary Contest is now active! Please declare your participation: https://forums.thedarkmod.com/index.php?/topic/22413-the-dark-mod-15th-anniversary-contest-entry-thread/
       
      · 0 replies
×
×
  • Create New...