1 #

2 # Cython -- Things that don't belong

3 # anywhere else in particular

4 #

5

6 import os, sys, re, codecs

7

8 def replace_suffix(path, newsuf):

9 base, _ = os.path.splitext(path)

10 return base + newsuf

11

12 def open_new_file(path):

13 if os.path.exists(path):

14 # Make sure to create a new file here so we can

15 # safely hard link the output files.

16 os.unlink(path)

17 return open(path, "w")

18

19 def castrate_file(path, st):

20 # Remove junk contents from an output file after a

21 # failed compilation.

22 # Also sets access and modification times back to

23 # those specified by st (a stat struct).

24 try:

25 f = open_new_file(path)

26 except EnvironmentError:

27 pass

28 else:

29 f.write(

30 "#error Do not use this file, it is the result of a failed Cython compilation.\n")

31 f.close()

32 if st:

33 os.utime(path, (st.st_atime, st.st_mtime-1))

34

35 def modification_time(path):

36 st = os.stat(path)

37 return st.st_mtime

38

39 def file_newer_than(path, time):

40 ftime = modification_time(path)

41 return ftime > time

42

43 # support for source file encoding detection

44

45 def encode_filename(filename):

46 if isinstance(filename, unicode):

47 return filename

48 try:

49 filename_encoding = sys.getfilesystemencoding()

50 if filename_encoding is None:

51 filename_encoding = sys.getdefaultencoding()

52 filename = filename.decode(filename_encoding)

53 except UnicodeDecodeError:

54 pass

55 return filename

56

57 _match_file_encoding = re.compile(u"coding[:=]\s*([-\w.]+)").search

58

59 def detect_file_encoding(source_filename):

60 # PEPs 263 and 3120

61 f = codecs.open(source_filename, "rU", encoding="UTF-8")

62 try:

63 chars = []

64 for i in range(2):

65 c = f.read(1)

66 while c and c != u'\n':

67 chars.append(c)

68 c = f.read(1)

69 encoding = _match_file_encoding(u''.join(chars))

70 if encoding:

71 return encoding.group(1)

72 finally:

73 f.close()

74 return "UTF-8"

75

76 def open_source_file(source_filename, mode="rU"):

77 encoding = detect_file_encoding(source_filename)

78 return codecs.open(source_filename, mode=mode, encoding=encoding)

79

80 def long_literal(value):

81 if isinstance(value, basestring):

82 if len(value) < 2:

83 value = int(value)

84 elif value[0] == 0:

85 value = int(value, 8)

86 elif value[1] in 'xX':

87 value = int(value[2:], 16)

88 else:

89 value = int(value)

90 return not -2**31 <= value < 2**31

91

92 # a simple class that simplifies the usage of utility code

93

94 class UtilityCode(object):

95 def __init__(self, proto=None, impl=None, init=None, cleanup=None):

96 self.proto = proto

97 self.impl = impl

98 self.init = init

99 self.cleanup = cleanup

100

101 def write_init_code(self, writer, pos):

102 if not self.init:

103 return

104 if callable(self.init):

105 self.init(writer, pos)

106 else:

107 writer.put(self.init)

108

109 def write_cleanup_code(self, writer, pos):

110 if not self.cleanup:

111 return

112 if callable(self.cleanup):

113 self.cleanup(writer, pos)

114 else:

115 writer.put(self.cleanup)