• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• fishroom-master
  • fishroom
    • telegram.py
    • gitter.py
    • db.py
    • photostore.py
    • xmpp.py
    • IRC.py
    • web
      • nickcolors.css
      • base.html
      • __main__.py
      • oauth.py
      • __init__.py
      • handlers.py
      • navbar.html
      • text_store.html
      • chat_log.html
    • telegram_tg.py
    • counter.py
    • textstore.py
    • fishroom.py
    • filestore.py
    • plugins
      • imglink.py
      • pia.py
      • hualao.py
      • stats.py
      • ratelimit.py
      • __init__.py
      • vote.py
    • models.py
    • dumpload.py
    • config.py.example
    • chatlogger.py
    • api_client.py
    • __init__.py
    • command.py
    • matrix.py
    • runner.py
    • bus.py
    • textformat.py
    • wechat.py
    • helpers.py
    • base.py
  • LICENSE
  • test
    • test_pastebin.py
    • __init__.py
    • test_vinergy.py
  • README.md
  • requirements.txt
  • Dockerfile
  • .gitignore

#!/usr/bin/env python3
from .photostore import BasePhotoStore
from io import BytesIO
import imghdr
import functools


class BaseFileStore(object):

    def upload_file(self, filename):
        raise Exception("Not Implemented")


class QiniuStore(BaseFileStore, BasePhotoStore):

    def __init__(self, access_key, secret_key, bucket_name, counter, base_url):
        try:
            import qiniu
        except ImportError:
            raise Exception("qiniu sdk is not installed")
        self.qiniu = qiniu

        self.auth = qiniu.Auth(access_key, secret_key)
        self.bucket = bucket_name
        self.counter = counter
        self.base_url = base_url

    def upload_image(self, filename=None, filedata=None, tag=None):
        token = self.auth.upload_token(self.bucket)
        if filedata is None:
            with open(filename, 'rb') as f:
                filedata = f.read()

        with BytesIO(filedata) as f:
            ext = imghdr.what(f)

        prefix = tag or "img"
        name = "%s/%02x.%s" % (prefix, self.counter.incr(), ext)

        ret, info = self.qiniu.put_data(token, name, filedata)
        if ret is None:
            return

        return self.base_url + name

    def upload_file(self, filedata, filename, filetype="file"):
        token = self.auth.upload_token(self.bucket)

        name = "%s/%02x-%s" % (filetype, self.counter.incr(), filename)
        ret, info = self.qiniu.put_data(token, name, filedata)
        if ret is None:
            return
        return self.base_url + name


def get_qiniu(redis_client, config):
    from .counter import Counter
    if 'qiniu' not in config:
        return None

    c = config['qiniu']
    counter = Counter(redis_client, 'qiniu')
    return QiniuStore(
        c['access_key'], c['secret_key'], c['bucket'],
        counter, c['base_url'],
    )


# vim: ts=4 sw=4 sts=4 expandtab