summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--Parser/tokenizer.c45
1 files changed, 30 insertions, 15 deletions
diff --git a/Parser/tokenizer.c b/Parser/tokenizer.c
index ff564ee..8497934 100644
--- a/Parser/tokenizer.c
+++ b/Parser/tokenizer.c
@@ -564,24 +564,39 @@ PyTokenizer_Get(tok, p_start, p_end)
/* Set start of current token */
tok->start = tok->cur - 1;
- /* Skip comment */
+ /* Skip comment, while looking for tab-setting magic */
if (c == '#') {
- /* Hack to allow overriding the tabsize in the file.
- This is also recognized by vi, when it occurs near the
- beginning or end of the file. (Will vi never die...?)
- For Python it must be at the beginning of the file! */
- /* XXX The real vi syntax is actually different :-( */
- /* XXX Should recognize Emacs syntax, too */
- int x;
- if (sscanf(tok->cur,
- " vi:set tabsize=%d:", &x) == 1 &&
- x >= 1 && x <= 40) {
- /* PySys_WriteStderr("# vi:set tabsize=%d:\n", x); */
- tok->tabsize = x;
- }
+ static char *tabforms[] = {
+ "tab-width:", /* Emacs */
+ ":tabstop=", /* vim, full form */
+ ":ts=", /* vim, abbreviated form */
+ "set tabsize=", /* will vi never die? */
+ /* more templates can be added here to support other editors */
+ };
+ char cbuf[80];
+ char *tp, **cp;
+ tp = cbuf;
do {
+ *tp++ = c = tok_nextc(tok);
+ } while (c != EOF && c != '\n' &&
+ tp - cbuf + 1 < sizeof(cbuf));
+ *tp = '\0';
+ for (cp = tabforms;
+ cp < tabforms + sizeof(tabforms)/sizeof(tabforms[0]);
+ cp++) {
+ if ((tp = strstr(cbuf, *cp))) {
+ int newsize = atoi(tp + strlen(*cp));
+
+ if (newsize >= 1 && newsize <= 40) {
+ tok->tabsize = newsize;
+ PySys_WriteStderr(
+ "Tab size set to %d\n",
+ newsize);
+ }
+ }
+ }
+ while (c != EOF && c != '\n')
c = tok_nextc(tok);
- } while (c != EOF && c != '\n');
}
/* Check for EOF and errors now */