Exemplo n.º 1
0
def oauth_authorized():

    remote_app = current_app.auth['app']

    # ripped off from Flask-OAuth
    # since `remote_app` should be defined in app start time
    if 'oauth_verifier' in request.args:
        data = remote_app.handle_oauth1_response()
    elif 'code' in request.args:
        data = remote_app.handle_oauth2_response()
    else:
        data = remote_app.handle_unknown_response()
    remote_app.free_request_token()
    # ---

    if data is None:
        return 'Access denied: reason=%s error=%s' % (
            request.args['error_reason'],
            request.args['error_description']
        )

    db = NodeIndex(current_app.config)

    session['oauth_token'] = (data['access_token'], '')

    user = current_app.auth['authenticator'].get_user_data()
    identity = current_app.auth['provider'] + ':' + str(user['id'])
    user_id = hashlib.sha1(identity).hexdigest()

    user_db = User.get(db.session, user_id)

    session['user_id'] = user_id
    session['user'] = user

    if not user_db:
        user_db = User(user_id=user_id, identity=identity,
                    secret_key=os.urandom(10).encode('hex'))
        db.add(user_db)

    db.query(User).filter_by(user_id=user_id).update({
        'name': user['name'],
        'picture': user['picture']
    })

    db.session.commit()

    broadcast('login', user_db.dict(private=False))

    next_page = request.args.get('next')

    if next_page:
        return redirect(next_page)
    else:
        return redirect(url_for('index'))
Exemplo n.º 2
0
class DistributedStorage(Storage):

    _songClass = RemoteSong

    def __init__(self, config):
        self.db = NodeIndex(config)
        self._config = config
        self._logger = logging.getLogger('radiocrepe.dist_storage')
        self.node_registry = NodeRegistry(self)

    def initialize(self):
        HubSideBase.metadata.create_all(self.db.engine)
        self.node_registry.detach_all()

    def mark_available(self, node_id):
        """
        Mark all songs stored in this node as available
        """
        self.db.query(self._songClass).filter_by(node_id=node_id).update({
            self._songClass.available: True
        })
        self.db.commit()

    def mark_unavailable(self, node_id):
        """
        Mark all songs in this node as unavailable
        """
        self.db.query(self._songClass).filter_by(node_id=node_id).update({
            self._songClass.available: False
        })
        self.db.commit()

    def file(self, uid):
        r = self._node_registry.get(uid)
        if r:
            return urlopen('http://%s/song/%s' % (r.server, r.uid))
        else:
            return None

    def update_node(self, node_id, data):
        for song in data:
            self._index(node_id=node_id, **song)

    @property
    def stats(self):
        return {
            "nodes": self.node_registry.stats
            }
Exemplo n.º 3
0
 def __init__(self, config):
     self.db = NodeIndex(config)
     self._config = config
     self._logger = logging.getLogger('radiocrepe.dist_storage')
     self.node_registry = NodeRegistry(self)