101 @classmethod |
101 @classmethod |
102 def pre_setup_database(cls, session, config): |
102 def pre_setup_database(cls, session, config): |
103 session.create_entity('CWSource', name=u'ldapuser', type=u'ldapfeed', parser=u'ldapfeed', |
103 session.create_entity('CWSource', name=u'ldapuser', type=u'ldapfeed', parser=u'ldapfeed', |
104 url=URL, config=CONFIG) |
104 url=URL, config=CONFIG) |
105 session.commit() |
105 session.commit() |
106 isession = session.repo.internal_session(safe=True) |
106 with session.repo.internal_session(safe=True) as isession: |
107 lfsource = isession.repo.sources_by_uri['ldapuser'] |
107 lfsource = isession.repo.sources_by_uri['ldapuser'] |
108 stats = lfsource.pull_data(isession, force=True, raise_on_error=True) |
108 stats = lfsource.pull_data(isession, force=True, raise_on_error=True) |
109 |
109 |
110 def _pull(self): |
110 def _pull(self): |
111 with self.session.repo.internal_session() as isession: |
111 with self.session.repo.internal_session() as isession: |
112 lfsource = isession.repo.sources_by_uri['ldapuser'] |
112 lfsource = isession.repo.sources_by_uri['ldapuser'] |
113 stats = lfsource.pull_data(isession, force=True, raise_on_error=True) |
113 stats = lfsource.pull_data(isession, force=True, raise_on_error=True) |