Adding a import_old_rp manager comand

apps/core/management/commands/import_old_rp.py

+import re
+
+import MySQLdb as ms
+from MySQLdb.cursors import DictCursor
+from django.core.management.base import BaseCommand
+
+from rp.models import Article
+
+
+class Command(BaseCommand):
+    help = """
+    Import data from the old press review. Should only be used by
+    LQDN staff since the dataformat of the old website only make
+    sense in their context.
+
+    It takes three arguments:
+        --host hostname
+        --sql_user username
+        --sql_password password
+    host defaults to 127.0.0.1.
+    """
+
+    def add_arguments(self, parser):
+        parser.add_argument('--host',
+                            default='127.0.0.1',
+                            dest='host',
+                            nargs='?')
+        parser.add_argument('--sql_user',
+                            dest='user',
+                            nargs='+')
+        parser.add_argument('--sql_password',
+                            dest='password',
+                            nargs='+')
+
+    def handle(self, *args, **options):
+        db = ms.connect(host=options['host'],
+                        user=options['user'][0],
+                        password=options['password'][0],
+                        db='site')
+        c = db.cursor(DictCursor)
+        # First, let's get the data from presse table
+        presse = c.execute("SELECT * FROM presse")
+
+        print("Importing 0/{} from previous database".format(presse))
+        # And here we go
+        done = 0
+        for item in c.fetchall():
+            done += 1
+            print("Importing {}/{} from previous database".format(done,
+                                                                  presse))
+            c.execute("SELECT nid, vid FROM node WHERE nid=%s", (item['nid'],))
+            node = c.fetchone()
+            if node is None:
+                continue
+
+            c.execute("SELECT body FROM node_revisions WHERE vid=%s",
+                      (node['vid'],))
+            revision = c.fetchone()
+            if revision is None:
+                continue
+
+            # Récupérons l'article si il existe en base
+            article = Article.add_new_url(url=item['url'])
+            if item['lang'] != "":
+                article.lang = item['lang']
+            article.published_at = item['date_publi']
+            article.title = item['title']
+
+            # Let's extract the website from the title
+            website = re.search(r'\[(.*)]', item['title'])
+            if website:
+                article.website = website.group(1)
+
+            # Augmentons le score si nécessaire
+            if item['note'] > 0:
+                article.und_score_up = item['note']
+            if item['note'] < 0:
+                article.und_score_down = abs(item['note'])
+            article.save()
+            article.refresh_from_db()
+
+            if item['published'] >= 1:
+                # Let's get the extracts
+                article.extracts = revision['body']
+                try:
+                    article.fetch_content()
+                    article.fetch_image()
+                except Exception:
+                    pass
+                if article.status not in ("DRAFT", "PUBLISHED", ):
+                    article.recover()
+
+            if item['published'] >= 2:
+                if article.status != "PUBLISHED":
+                    article.publish()
+
+            article.save()