Switch to unified view
a/scripts/restore_project.py | b/scripts/restore_project.py | ||
---|---|---|---|
... |
... |
||
62 | 62 | ||
63 | def get_repo_collections(): |
63 | def get_repo_collections(): |
64 | res = {} |
64 | res = {} |
65 | for m in Mapper.all_mappers(): |
65 | for m in Mapper.all_mappers(): |
66 | cls = m.mapped_class |
66 | cls = m.mapped_class |
67 | cname = cls.__mongometa__.name |
67 | cname = m.collection.m.collection_name |
68 | if issubclass(cls, M.Repository): res[cname] = cls |
68 | if issubclass(cls, M.Repository): res[cname] = cls |
69 | return res |
69 | return res |
70 | 70 | ||
71 | def _read_bson(fp): |
71 | def _read_bson(fp): |
72 | slen = fp.read(4) |
72 | slen = fp.read(4) |