added missing case to get_docstring
diff --git a/Python/compile.c b/Python/compile.c
index 93d5b6d..45bed39 100644
--- a/Python/compile.c
+++ b/Python/compile.c
@@ -483,7 +483,6 @@
int op;
node *n;
{
- object *v;
char *name;
char buffer[1000];
/* XXX it is possible to write this code without the 1000
@@ -1868,13 +1867,14 @@
get_docstring(n)
node *n;
{
+ int i;
+
switch (TYPE(n)) {
case suite:
if (NCH(n) == 1)
return get_docstring(CHILD(n, 0));
else {
- int i;
for (i = 0; i < NCH(n); i++) {
node *ch = CHILD(n, i);
if (TYPE(ch) == stmt)
@@ -1883,6 +1883,14 @@
}
break;
+ case file_input:
+ for (i = 0; i < NCH(n); i++) {
+ node *ch = CHILD(n, i);
+ if (TYPE(ch) == stmt)
+ return get_docstring(ch);
+ }
+ break;
+
case stmt:
case simple_stmt:
case small_stmt: