def _handle(self, *args, **options): self.verbosity = int(options.get('verbosity')) self.interactive = options.get('interactive') only_users = options.get("only_users", False) flush_djangobb = options.get("flush_djangobb", False) if flush_djangobb: self._flush_djangobb() return self.max_entries = options.get("max_entries") if self.max_entries: self.warn("Migrate with max entries: %i !\n" % self.max_entries) self.check_attachment_path() self.check_models() # disable DjangoBB signals for speedup post_save.disconnect(djangobb_signals.post_saved, sender=Post, dispatch_uid='djangobb_post_save') post_save.disconnect(djangobb_signals.topic_saved, sender=Topic, dispatch_uid='djangobb_topic_save') # Speedup migration by disable haystack search index creation. site.unregister(Post) #disable_auto_fields(Forum) disable_auto_fields(Topic) disable_auto_fields(Post) cleanup_users = int(options.get("cleanup_users")) moderator_groups = phpbb_Group.objects.filter( name__in=["ADMINISTRATORS", "GLOBAL_MODERATORS"]) user_dict, moderators = self.migrate_users(cleanup_users, moderator_groups) if only_users: self.out(u"\nmigrate only users done.\n") return forum_dict = self.migrate_forums(moderators) self.migrate_topic(user_dict, forum_dict) self.migrate_posts(user_dict) # needed if signals disabled, see above self.update_topic_stats() self.update_forum_stats() self.out(u"\nmigration done.\n")
def _handle(self, *args, **options): self.verbosity = int(options.get('verbosity')) self.interactive = options.get('interactive') only_users = options.get("only_users", False) flush_djangobb = options.get("flush_djangobb", False) if flush_djangobb: self._flush_djangobb() return self.max_entries = options.get("max_entries") if self.max_entries: self.warn("Migrate with max entries: %i !\n" % self.max_entries) self.check_attachment_path() self.check_models() # disable DjangoBB signals for speedup post_save.disconnect(djangobb_signals.post_saved, sender=Post, dispatch_uid='djangobb_post_save') post_save.disconnect(djangobb_signals.topic_saved, sender=Topic, dispatch_uid='djangobb_topic_save') # Speedup migration by disable haystack search index creation. site.unregister(Post) #disable_auto_fields(Forum) disable_auto_fields(Topic) disable_auto_fields(Post) cleanup_users = int(options.get("cleanup_users")) moderator_groups = phpbb_Group.objects.filter( name__in=["ADMINISTRATORS", "GLOBAL_MODERATORS"] ) user_dict, moderators = self.migrate_users(cleanup_users, moderator_groups) if only_users: self.out(u"\nmigrate only users done.\n") return forum_dict = self.migrate_forums(moderators) self.migrate_topic(user_dict, forum_dict) self.migrate_posts(user_dict) # needed if signals disabled, see above self.update_topic_stats() self.update_forum_stats() self.out(u"\nmigration done.\n")
def test_registrations(self): from haystack import backend sb = backend.SearchBackend() self.assertEqual(len(sb.site.get_indexed_models()), 2) from haystack import site self.assertEqual(len(site.get_indexed_models()), 2) from site_registration.models import Foo, Bar site.unregister(Bar) self.assertEqual(len(sb.site.get_indexed_models()), 1) self.assertEqual(len(site.get_indexed_models()), 1) site.unregister(Foo) self.assertEqual(len(sb.site.get_indexed_models()), 0) self.assertEqual(len(site.get_indexed_models()), 0)