# -*- coding: utf-8 -*- from datetime import datetime import json import os import shutil import ox from sqlalchemy.orm import load_only import sqlalchemy as sa from changelog import add_record from db import MutableDict import db import json_pickler import settings import state import utils import media from websocket import trigger_event import logging logger = logging.getLogger(__name__) class User(db.Model): __tablename__ = 'user' created = sa.Column(sa.DateTime()) modified = sa.Column(sa.DateTime()) id = sa.Column(sa.String(43), primary_key=True) info = sa.Column(MutableDict.as_mutable(sa.PickleType(pickler=json_pickler))) nickname = sa.Column(sa.String(256), unique=True) pending = sa.Column(sa.String(64)) # sent|received queued = sa.Column(sa.Boolean()) peered = sa.Column(sa.Boolean()) online = sa.Column(sa.Boolean()) def __repr__(self): return self.id @classmethod def get(cls, id, for_update=False): qs = cls.query.filter_by(id=id) if for_update: qs = qs.with_for_update() user = qs.first() if user and not user.info: user.info = {} return user @classmethod def get_or_create(cls, id): user = cls.get(id) if not user: user = cls(id=id, peered=False, online=False) user.info = {} if state.nodes and state.nodes.local and id in state.nodes.local: user.info['local'] = state.nodes.local[id] user.info['username'] = user.info['local']['username'] user.update_name() user.save() return user def save(self): state.db.session.add(self) state.db.session.commit() @property def name(self): name = self.nickname if self.id != settings.USER_ID else '' return name @property def library(self): l = List.get_or_create(self.id, '') if l.index_ != -1: l.index_ = -1 l.save() return l def json(self, keys=None): j = {} if self.info: j.update(self.info) j['id'] = self.id if self.pending: j['pending'] = self.pending j['peered'] = self.peered if not keys or 'online' in keys: j['online'] = self.is_online() j['name'] = self.name if not keys or 'username' in keys or 'contact' in keys: if self.id == settings.USER_ID: j['username'] = settings.preferences['username'] j['contact'] = settings.preferences['contact'] elif self.id in state.peers: peer = state.peers[self.id] for key in ('username', 'contact'): if key in peer.info: j[key] = peer.info[key] if keys: for k in set(j) - set(keys): del j[k] return j def export_library(self): old_path = os.path.join(os.path.expanduser(settings.preferences['libraryPath']), 'Books', 'library.json') if os.path.exists(old_path): os.unlink(old_path) path = os.path.join(settings.data_path, 'library.json') self.library.export_json(path) def is_online(self): return state.nodes and state.nodes.is_online(self.id) def trigger_status(self): trigger_event('status', { 'id': self.id, 'online': self.is_online() }) def lists_json(self): self.library if self.id != settings.USER_ID: peer = utils.get_peer(self.id) lists = [] lists.append({ 'id': self.nickname + ':', 'user': self.name, 'items': len(peer.library), 'name': 'Library', 'type': 'library' }) index = 0 for name in peer.info.get('listorder', peer.info.get('lists', []).keys()): lists.append({ 'id': '%s:%s' % (self.nickname, name), 'user': self.name, 'name': name, 'index': index, 'items': len(peer.info['lists'].get(name, [])), 'type': 'static' }) index += 1 return lists return [l.json() for l in self.lists.order_by('index_')] def update_peering(self, peered, username=None): was_peering = self.peered if peered: logging.debug('update_peering, pending: %s queued: %s', self.pending, self.queued) self.queued = self.pending != 'sent' self.pending = '' if username: self.info['username'] = username self.update_name() if not was_peering: add_record('addpeer', self.id, self.nickname) if 'index' not in self.info: self.info['index'] = max([ u.info.get('index', -1) for u in User.query.filter_by(peered=True) if u.id != self.id ] + [0]) + 1 self.peered = True self.save() if self.id in state.removepeer: del state.removepeer[self.id] else: self.pending = '' self.peered = False self.queued = False if 'index' in self.info: del self.info['index'] self.update_name() self.save() if self.name in settings.ui['showFolder']: del settings.ui['showFolder'][self.name] settings.ui._save() state.removepeer[self.id] = True self.cleanup() if was_peering: add_record('removepeer', self.id) self.save() def cleanup(self): from item.models import user_items, Item List.query.filter_by(user_id=self.id).delete() c_user_id = user_items.columns['user_id'] q = user_items.delete().where(c_user_id.is_(self.id)) state.db.session.execute(q) Item.remove_without_user() self.save() if self.id in state.peers: state.peers[self.id].remove() del state.peers[self.id] def update_name(self): if self.id == settings.USER_ID: name = settings.preferences.get('username', 'anonymous') else: name = self.info.get('nickname') or self.info.get('username') or 'anonymous' nickname = name n = 2 while self.query.filter_by(nickname=nickname).filter(User.id != self.id).first(): nickname = '%s [%d]' % (name, n) n += 1 self.nickname = nickname def rebuild_changelog(self): logger.error('no longer used') return Changelog.query.filter_by(user_id=self.id).delete() for item in self.library.get_items().order_by('created'): Changelog.record(self, 'additem', item.id, item.info, _commit=False) Changelog.record(self, 'edititem', item.id, item.meta, _commit=False) lists = [] for l in List.query.filter_by(user_id=self.id, type='static').order_by('index_'): if l.name: lists.append(l.name) Changelog.record(self, 'addlist', l.name, _commit=False) items = [i.id for i in l.get_items().options(load_only('id'))] if items: Changelog.record(self, 'addlistitems', l.name, items, _commit=False) if len(lists) > 1: Changelog.record(self, 'orderlists', lists, _commit=False) for peer in User.query.filter_by(peered=True): Changelog.record(self, 'addpeer', peer.id, self.nickname, _commit=False) if peer.info.get('contact'): Changelog.record(self, 'editpeer', peer.id, { 'contact': peer.info.get('contact') }, _commit=False) if settings.preferences.get('contact'): Changelog.record(self, 'editcontact', settings.preferences.get('contact'), _commit=False) state.db.session.commit() list_items = sa.Table('listitem', db.metadata, sa.Column('list_id', sa.Integer(), sa.ForeignKey('list.id')), sa.Column('item_id', sa.String(32), sa.ForeignKey('item.id'))) class List(db.Model): __tablename__ = 'list' id = sa.Column(sa.Integer(), primary_key=True) name = sa.Column(sa.String()) index_ = sa.Column(sa.Integer()) type = sa.Column(sa.String(64)) _query = sa.Column('query', MutableDict.as_mutable(sa.PickleType(pickler=json_pickler))) user_id = sa.Column(sa.String(43), sa.ForeignKey('user.id')) user = sa.orm.relationship('User', backref=sa.orm.backref('lists', lazy='dynamic')) items = sa.orm.relationship('Item', secondary=list_items, backref=sa.orm.backref('lists', lazy='dynamic')) @classmethod def get(cls, user_id, name=None): if name is None: user_id, name = cls.get_user_name(user_id) return cls.query.filter_by(user_id=user_id, name=name).first() @classmethod def get_user_name(cls, user_id): nickname, name = user_id.split(':', 1) if nickname: user = User.query.filter_by(nickname=nickname).first() user_id = user.id else: user_id = settings.USER_ID return user_id, name @classmethod def get_or_create(cls, user_id, name=None, query=None): if name is None: user_id, name = cls.get_user_name(user_id) l = cls.get(user_id, name) if not l: l = cls.create(user_id, name, query) return l @classmethod def create(cls, user_id, name, query=None): prefix = name n = 2 while cls.get(user_id, name): name = '%s [%s]' % (prefix, n) n += 1 l = cls(user_id=user_id, name=name) if query: l._query = query l.type = 'smart' else: l.type = 'static' l.index_ = cls.query.filter_by(user_id=user_id).count() state.db.session.add(l) state.db.session.commit() if user_id == settings.USER_ID: if l.type == 'static' and name != '': add_record('addlist', l.name) return l @classmethod def rename_user(cls, old, new): for l in cls.query.filter(cls._query != None): def update_conditions(conditions): changed = False for c in conditions: if 'conditions' in c: changed = update_conditions(c['conditions']) else: if c.get('key') == 'list' and c.get('value', '').startswith('%s:' % old): c['value'] = '%s:%s' % new, c['value'].split(':', 1)[1] changed = True return changed if l._query and update_conditions(l._query.get('conditions', [])): l.save() def add_items(self, items, commit=True): from item.models import Item available_items = [] for item_id in items: i = Item.get(item_id) if i: if self.user_id == settings.USER_ID and i.info.get('mediastate') != 'available': i.queue_download() if i not in self.items: self.items.append(i) i.update(commit=False) if i.info['mediastate'] == 'available': available_items.append(item_id) state.db.session.add(self) if commit: state.db.session.commit() if self.user_id == settings.USER_ID and self.name != '' and available_items: add_record('addlistitems', self.name, available_items) def get_items(self): from item.models import Item if self.type == 'smart': return Item.find({'query': self._query}) else: return self.user.items.join(Item.lists, aliased=True).filter(List.id == self.id) def remove_items(self, items, commit=True): from item.models import Item for item_id in items: i = Item.get(item_id) if i: if i in self.items: self.items.remove(i) i.update(commit=commit) state.db.session.add(self) if commit: state.db.session.commit() if self.user_id == settings.USER_ID and self.name != '': add_record('removelistitems', self.name, items) def remove(self, commit=True): if not self._query: q = list_items.delete().where(list_items.columns['list_id'].is_(self.id)) state.db.session.execute(q) if not self._query: if self.user_id == settings.USER_ID and self.name != '': add_record('removelist', self.name) state.db.session.delete(self) if commit: state.db.session.commit() @property def public_id(self): id = '' if self.user_id != settings.USER_ID: id += self.user.nickname id = '%s:%s' % (id, self.name) return id @property def find_id(self): id = '' if self.user_id != settings.USER_ID: id += self.user_id id = '%s:%s' % (id, self.id) return id def __repr__(self): return self.public_id def items_count(self): if self.user_id != settings.USER_ID: peer = utils.get_peer(self.user_id) if self.name: return len(peer.info['lists'].get(self.name, [])) else: return len(peer.library) return self.get_items().count() def json(self): r = { 'id': self.public_id, 'user': self.user.name, 'name': self.name, 'index': self.index_, 'items': self.items_count(), 'type': self.type } if self.name == '': r['name'] = 'Library' r['type'] = 'library' del r['index'] if self.type == 'smart': r['query'] = self._query return r def save(self): state.db.session.add(self) state.db.session.commit() def create_symlinks(self): pass def export_json(self, path=None): from utils import _to_json if not path: if self.name: name = os.path.join('Lists', self.name) else: name = 'Books' path = os.path.join(os.path.expanduser(settings.preferences['libraryPath']), name, 'library.json') ox.makedirs(os.path.dirname(path)) items = [] for i in self.get_items(): j = i.json() for f in i.files: j['path'] = f.path break items.append(j) with open(path, 'w', encoding='utf-8') as f: json.dump(items, f, indent=4, default=_to_json, ensure_ascii=False, sort_keys=True) class Metadata(db.Model): __tablename__ = 'user_metadata' created = sa.Column(sa.DateTime()) modified = sa.Column(sa.DateTime()) id = sa.Column(sa.Integer(), primary_key=True) item_id = sa.Column(sa.String(32)) user_id = sa.Column(sa.String(43), sa.ForeignKey('user.id')) data_hash = sa.Column(sa.String(40), index=True) data = sa.Column(MutableDict.as_mutable(sa.PickleType(pickler=json_pickler))) def __repr__(self): return '{item}/{user}'.format(item=self.item_id, user=self.user_id) @property def timestamp(self): return utils.datetime2ts(self.modified) @classmethod def get(cls, user_id, item_id): return cls.query.filter_by(item_id=item_id, user_id=user_id).first() @classmethod def get_or_create(cls, user_id, item_id, data=None, commit=True): m = cls.get(user_id=user_id, item_id=item_id) if not m: m = cls(user_id=user_id, item_id=item_id) m.created = datetime.utcnow() if data: m.data = data else: m.data = {} m.save(commit=commit) elif data: m.edit(data, commit=commit) return m def get_hash(self): return utils.get_meta_hash(self.data) def save(self, commit=True, modified=None): if modified is None: self.modified = datetime.utcnow() else: self.modified = modified state.db.session.add(self) if commit: state.db.session.commit() def edit(self, data, commit=True, modified=None): changes = {} if 'isbn' in data and isinstance(data['isbn'], list): isbns = [utils.to_isbn13(isbn) for isbn in data['isbn']] isbns = [isbn for isbn in isbns if isbn] if isbns: data['isbn'] = isbns[0] else: del data['isbn'] for key in data: if key == 'id': continue if data[key] != self.data.get(key): self.data[key] = data[key] changes[key] = data[key] if changes: self.data_hash = self.get_hash() self.save(commit=commit, modified=modified) return changes def delete(self): state.db.session.delete(self) state.db.session.commit() def export_list(data): with db.session(): self = List.get(data['list']) if not self: return mode = data.get('mode') prefix = data.get('path') if mode not in ('add', 'replace'): logger.debug('invalid mode %s', mode) return if not prefix or prefix == '/': logger.debug('invalid export path %s', prefix) trigger_event('activity', { 'activity': 'export', 'path': prefix, 'progress': [0, 0], 'status': {'code': 404, 'text': 'invalid export path'} }) return root = prefix while not os.path.exists(root) and root != '/': root = os.path.dirname(root) if not os.access(root, os.W_OK): logger.debug('can not write to %s', root) trigger_event('activity', { 'activity': 'export', 'path': prefix, 'progress': [0, 0], 'path': prefix, 'status': {'code': 404, 'text': 'permission denied'} }) return if os.path.exists(prefix): existing_files = set( os.path.join(root, f) for root, _, files in os.walk(prefix) for f in files ) else: existing_files = set() new_files = set() count = self.get_items().count() n = 1 for i in self.get_items(): if i.files.all(): f = i.files.all()[0] source = f.fullpath() target = os.path.join(prefix, f.path) if mode == 'add': p = 1 parts = target.rsplit('.', 1) while os.path.exists(target) and media.get_id(target) != f.sha1: target = '.'.join([parts[0], f.sha1[:p], parts[1]]) p += 1 ox.makedirs(os.path.dirname(target)) if os.path.exists(target): if mode == 'replace' and media.get_id(target) != f.sha1: os.unlink(target) shutil.copy2(source, target) else: shutil.copy2(source, target) new_files.add(target) trigger_event('activity', { 'activity': 'export', 'path': prefix, 'progress': [n, count] }) n += 1 if mode == 'replace': for f in list(existing_files - new_files): os.unlink(f) utils.remove_empty_folders(prefix) self.export_json(os.path.join(prefix, 'library.json')) trigger_event('activity', { 'activity': 'export', 'progress': [count, count], 'path': prefix, 'status': {'code': 200, 'text': ''}, }) def update_user_peering(user_id, peered, username=None): with db.session(): u = User.get(user_id, for_update=True) if u: u.update_peering(peered, username) def remove_local_info(id): with db.session(): u = User.get(id, for_update=True) if u and 'local' in u.info: del u.info['local'] u.save() u.trigger_status() def add_local_info(data): with db.session(): u = User.get(data['id'], for_update=True) if u: if u.info['username'] != data['username']: u.info['username'] = data['username'] u.update_name() u.info['local'] = data u.save() if state.nodes: state.nodes.queue('add', u.id) def upload(data): delay = 60 with db.session(): u = User.get(data['user']) if u: if u.is_online() and state.nodes._nodes[u.id].upload(data['items']): pass else: state.main.call_later(delay, lambda: state.tasks.queue('upload', data))