spradio-server-django/savepointradio/radio/management/commands/importradio.py

from decimal import *
import os
import sqlite3

from django.core.management.base import BaseCommand, CommandError

from radio.models import Album, Artist, Game, Song

getcontext().prec = 8


def adapt_decimal(d):
    return str(d)


def convert_decimal(s):
    return Decimal(s.decode('utf8'))


sqlite3.register_adapter(Decimal, adapt_decimal)
sqlite3.register_converter("decimal", convert_decimal)


class Command(BaseCommand):
    help = 'Imports the old radio data from the original sqlite3 database'

    def add_arguments(self, parser):
        parser.add_argument('sqlite3_db_file', nargs=1)

    def handle(self, *args, **options):
        if not os.path.isfile(options['sqlite3_db_file'][0]):
            raise CommandError('File does not exist')
        else:
            total_albums = 0
            total_artists = 0
            total_games = 0
            total_songs = 0
            total_jingles = 0

            detect_types = sqlite3.PARSE_DECLTYPES | sqlite3.PARSE_COLNAMES
            con = sqlite3.connect(options['sqlite3_db_file'][0],
                                  detect_types=detect_types)
            cur = con.cursor()

            # Fetching albums first
            for album in con.execute('SELECT title, enabled FROM albums'):
                album_disabled = not bool(album[1])
                Album.objects.create(title=album[0], disabled=album_disabled)
                total_albums += 1

            self.stdout.write('Imported {} albums'.format(str(total_albums)))

            # Next up, artists
            cur.execute('''SELECT
                            artists_id,
                            alias,
                            firstname,
                            lastname,
                            enabled
                        FROM artists''')
            artists = cur.fetchall()

            for artist in artists:
                artist_disabled = not bool(artist[4])
                Artist.objects.create(alias=artist[1] or '',
                                      first_name=artist[2] or '',
                                      last_name=artist[3] or '',
                                      disabled=artist_disabled)
                total_artists += 1

            self.stdout.write('Imported {} artists'.format(str(total_artists)))

            # On to games
            for game in con.execute('SELECT title, enabled FROM games'):
                game_disabled = not bool(game[1])
                Game.objects.create(title=game[0], disabled=game_disabled)
                total_games += 1

            self.stdout.write('Imported {} games'.format(str(total_games)))

            # Followed by the songs
            cur.execute('''SELECT
                            songs.songs_id AS id,
                            games.title AS game,
                            albums.title AS album,
                            songs.enabled AS enabled,
                            songs.type AS type,
                            songs.title AS title,
                            songs.length AS length,
                            songs.path AS path
                        FROM songs
                        LEFT JOIN games
                            ON (songs.game = games.games_id)
                        LEFT JOIN albums
                            ON (songs.album = albums.albums_id)''')
            songs = cur.fetchall()

            for song in songs:
                try:
                    album = Album.objects.get(title__exact=song[2])
                except Album.DoesNotExist:
                    album = None

                try:
                    game = Game.objects.get(title__exact=song[1])
                except Game.DoesNotExist:
                    game = None

                song_disabled = not bool(song[3])
                new_song = Song.objects.create(album=album,
                                               game=game,
                                               disabled=song_disabled,
                                               song_type=song[4],
                                               title=song[5],
                                               length=song[6],
                                               path=song[7])
                if song[4] == 'S':
                    total_songs += 1
                else:
                    total_jingles += 1

                cur.execute('''SELECT
                                ifnull(alias, "") AS alias,
                                ifnull(firstname, "") AS firstname,
                                ifnull(lastname, "") AS lastname
                            FROM artists
                            WHERE artists_id
                            IN (SELECT artists_artists_id
                                FROM songs_have_artists
                                WHERE songs_songs_id = ?)''', [song[0]])
                old_artists = cur.fetchall()
                for old_artist in old_artists:
                    new_artist = Artist.objects.get(alias__exact=old_artist[0],
                                                    first_name__exact=old_artist[1],
                                                    last_name__exact=old_artist[2])
                    new_song.artists.add(new_artist)

            self.stdout.write(
                'Imported {} songs ({} requestables, {} jingles)'.format(
                    str(total_songs + total_jingles),
                    str(total_songs),
                    str(total_jingles)
                )
            )

            pub = input('Do you want to publish all imported objects as well? '
                        '[Y/N] ')

            if pub == 'Y' or pub == 'y':
                for al in Album.objects.all():
                    al.publish()
                for ar in Artist.objects.all():
                    ar.publish()
                for g in Game.objects.all():
                    g.publish()
                for s in Song.objects.all():
                    s.publish()
                self.stdout.write('Published imported objects successfully')
            else:
                self.stdout.write('Skipped publishing songs')