library.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
#
#  Copyright (C) 2012  James Adams
#
#  This program is free software; you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation; either version 2 of the License, or
#  (at your option) any later version.
#
#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#
#  You should have received a copy of the GNU General Public License
#  along with this program; if not, write to the Free Software
#  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
#  MA 02110-1301, USA.
#


from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, String, DateTime
from sqlalchemy.sql.expression import func
from datetime import datetime
from os.path import dirname
from sys import argv

Base = declarative_base()

SCHEMA_VERSION = '2'

class sspTrack(Base):
    __tablename__ = 'tracks'
    filepath = Column(String(512), primary_key=True)
    playcount = Column(Integer())
    skipcount = Column(Integer())
    lastplayed = Column(DateTime())
    albumid = Column(String(48))
    trackid = Column(String(48))
    lastscanned = Column(DateTime())

    def __init__(self, filepath):
        self.filepath = filepath
        self.playcount = 0
        self.skipcount = 0
        self.lastplayed = None
        self.albumid = ""
        self.trackid = ""
        self.lastscanned = None

    def __repr__(self):
        return "<Track (%s - %s plays, %s skips, last played %s, last scanned %s)>" % (
            self.filepath,
            self.playcount,
            self.skipcount,
            self.lastplayed,
            self.lastscanned,
        )


class sspStat(Base):
    __tablename__ = 'stats'
    hour = Column(Integer(), primary_key=True)
    playcount = Column(Integer())
    skipcount = Column(Integer())

    def __init__(self, hour):
        if hour >= 0 and hour < 24:
            self.hour = hour
            self.playcount = 0
            self.skipcount = 0

    def __repr__(self):
        return "<Stat (Hour %s - %s plays, %s skips)>" % (self.hour, self.playcount, self.skipcount)


class sspWeekStat(Base):
    __tablename__ = 'weekstats'
    hour = Column(Integer(), primary_key=True)
    day = Column(Integer(), primary_key=True)
    playcount = Column(Integer())
    skipcount = Column(Integer())

    def __init__(self, hour, day):
        if hour >= 0 and hour < 24 and day >= 0 and day < 7:
            self.hour = hour
            self.day = day
            self.playcount = 0
            self.skipcount = 0

    def __repr__(self):
        return "<WeekStat (Hour %s, Day %s - %s plays, %s skips)>" % (
            self.hour,
            self.day,
            self.playcount,
            self.skipcount,
        )


def connect():
    base_path = dirname(argv[0])
    engine = create_engine('sqlite:///%s/library-v%s.db' % (base_path, SCHEMA_VERSION), echo=False)

    Base.metadata.create_all(engine)

    Session = sessionmaker(bind=engine)

    return(Session())