diff options
Diffstat (limited to 'lib/wind/generate.py')
| -rw-r--r-- | lib/wind/generate.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/lib/wind/generate.py b/lib/wind/generate.py index 1d708c6a3c47..0a32fcd4d4d8 100644 --- a/lib/wind/generate.py +++ b/lib/wind/generate.py @@ -37,6 +37,7 @@ import datetime import string import os +import sys class GeneratedFile : "Represents a generated file" @@ -56,10 +57,14 @@ class GeneratedFile : class Header(GeneratedFile) : "Represents a generated header file" - guardTrans = string.maketrans('-.', '__') + if sys.hexversion>0x300000f: + guardTrans = bytes.maketrans(b'-.', b'__') + else: + import string + guardTrans = string.maketrans('-.', '__') def makeGuard(self) : """Return a name to be used as ifdef guard""" - return string.upper(string.translate(self.name, self.guardTrans)) + return self.name.translate(self.guardTrans).upper() def __init__(self, name) : "Create a new Header with name" |