mirror of
https://github.com/colbymchenry/codegraph.git
synced 2026-10-17 05:07:37 +08:00
Colby Mchenry
34ede4de88
resolvePythonModuleMember (`mod.member` after `import mod` / `from pkg import mod`) looks the member up with pythonModuleSymbol, so a package's __init__.py re-exports lead to the definition. Co-authored-by: Claude Opus 5.5 <noreply@anthropic.com>
84 lines
2.8 KiB
TypeScript
84 lines
2.8 KiB
TypeScript
/**
|
|
* A Python name imported from a package resolves through the package's
|
|
* re-exports — `from .users import *`, `from .tokens import Token` in its
|
|
* `__init__.py` — to where it is defined. netbox's `from users.models import
|
|
* User` names `users/models/__init__.py`, which star-imports `.users`;
|
|
* `httpx.Client(…)` names `httpx/__init__.py`, which imports `._client`.
|
|
*/
|
|
import { describe, it, expect, afterAll, beforeAll } from 'vitest';
|
|
import * as fs from 'fs';
|
|
import * as os from 'os';
|
|
import * as path from 'path';
|
|
import { CodeGraph } from '../src';
|
|
|
|
let root = '';
|
|
let cg: CodeGraph;
|
|
|
|
beforeAll(async () => {
|
|
root = fs.mkdtempSync(path.join(os.tmpdir(), 'cg-py-reexport-'));
|
|
const files: Record<string, string> = {
|
|
'netbox/users/__init__.py': '',
|
|
'netbox/users/models/__init__.py': `from .tokens import Token
|
|
from .users import *
|
|
`,
|
|
'netbox/users/models/users.py': `class User:
|
|
pass
|
|
`,
|
|
'netbox/users/models/tokens.py': `class Token:
|
|
pass
|
|
`,
|
|
'netbox/extras/tests/test_templatetags.py': `class User:
|
|
pass
|
|
`,
|
|
'netbox/alembic/versions/0001_tokens.py': `class Token:
|
|
pass
|
|
`,
|
|
'netbox/netbox/__init__.py': '',
|
|
'netbox/netbox/views/__init__.py': '',
|
|
'netbox/netbox/views/generic/__init__.py': `from .object_views import *
|
|
`,
|
|
'netbox/netbox/views/generic/object_views.py': `class ObjectEditView:
|
|
pass
|
|
`,
|
|
'netbox/dcim/views.py': `from netbox.views import generic
|
|
|
|
|
|
class DeviceEditView(generic.ObjectEditView):
|
|
pass
|
|
`,
|
|
'netbox/extras/views.py': `class ObjectEditView:
|
|
pass
|
|
`,
|
|
'netbox/core/views.py': `from users.models import Token, User
|
|
|
|
|
|
def make():
|
|
return User(), Token()
|
|
`,
|
|
};
|
|
for (const [rel, content] of Object.entries(files)) {
|
|
fs.mkdirSync(path.dirname(path.join(root, rel)), { recursive: true });
|
|
fs.writeFileSync(path.join(root, rel), content);
|
|
}
|
|
cg = await CodeGraph.init(root, { index: true });
|
|
});
|
|
|
|
afterAll(() => {
|
|
cg?.close();
|
|
if (root) fs.rmSync(root, { recursive: true, force: true });
|
|
});
|
|
|
|
describe('Python package re-exports', () => {
|
|
it('lead a module member to its definition', () => {
|
|
const ids = cg.getNodesInFile('netbox/dcim/views.py').map((n) => n.id);
|
|
const bases = cg.getOutgoingEdgesFrom(ids).filter((e) => e.kind === 'extends').map((e) => cg.getNode(e.target)!.filePath);
|
|
expect(bases).toEqual(['netbox/netbox/views/generic/object_views.py']);
|
|
});
|
|
|
|
it('lead an imported name to its definition, through star and named re-exports', () => {
|
|
const ids = cg.getNodesInFile('netbox/core/views.py').map((n) => n.id);
|
|
const targets = cg.getOutgoingEdgesFrom(ids).filter((e) => e.kind === 'instantiates').map((e) => cg.getNode(e.target)!.filePath).sort();
|
|
expect(targets).toEqual(['netbox/users/models/tokens.py', 'netbox/users/models/users.py']);
|
|
});
|
|
});
|