Skip to content

Commit d466d2b

Browse files
authored
bugfix for issue #174 (#191)
"auto()" suggests int enums, but pyo3 does not offer them, therefore use "..." instead this fixed issue #174, if you agree with the statement written there, please merge my change.
1 parent 236ed7a commit d466d2b

File tree

3 files changed

+7
-7
lines changed

3 files changed

+7
-7
lines changed

examples/pure/pure.pyi

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ import datetime
66
import os
77
import pathlib
88
import typing
9-
from enum import Enum, auto
9+
from enum import Enum
1010

1111
MY_CONSTANT: builtins.int
1212
class A:
@@ -25,8 +25,8 @@ class MyDate(datetime.date):
2525
...
2626

2727
class Number(Enum):
28-
FLOAT = auto()
29-
INTEGER = auto()
28+
FLOAT = ...
29+
INTEGER = ...
3030

3131
is_float: builtins.bool
3232
r"""
@@ -39,8 +39,8 @@ class Number(Enum):
3939
"""
4040

4141
class NumberRenameAll(Enum):
42-
FLOAT = auto()
43-
INTEGER = auto()
42+
FLOAT = ...
43+
INTEGER = ...
4444

4545
def ahash_dict() -> builtins.dict[builtins.str, builtins.int]: ...
4646

pyo3-stub-gen/src/generate/enum_.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ impl fmt::Display for EnumDef {
2929
let indent = indent();
3030
docstring::write_docstring(f, self.doc, indent)?;
3131
for variants in self.variants {
32-
writeln!(f, "{indent}{} = auto()", variants)?;
32+
writeln!(f, "{indent}{} = ...", variants)?;
3333
}
3434
for member in &self.members {
3535
writeln!(f)?;

pyo3-stub-gen/src/generate/module.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ impl fmt::Display for Module {
4848
writeln!(f, "from . import {}", submod)?;
4949
}
5050
if !self.enum_.is_empty() {
51-
writeln!(f, "from enum import Enum, auto")?;
51+
writeln!(f, "from enum import Enum")?;
5252
}
5353
writeln!(f)?;
5454

0 commit comments

Comments
 (0)