You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
1.3 KiB
60 lines
1.3 KiB
8 years ago
|
from flask import Flask, g
|
||
|
|
from cookiespool.config import *
|
||
|
|
from cookiespool.db import CookiesRedisClient, AccountRedisClient
|
||
|
|
|
||
|
|
__all__ = ['app']
|
||
|
|
|
||
|
|
app = Flask(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
@app.route('/')
|
||
|
|
def index():
|
||
|
|
return '<h2>Welcome to Cookie Pool System</h2>'
|
||
|
|
|
||
|
|
|
||
|
|
def get_conn():
|
||
|
|
"""
|
||
|
|
获取
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
for name in GENERATOR_MAP:
|
||
|
|
print(name)
|
||
|
|
if not hasattr(g, name):
|
||
|
|
setattr(g, name + '_cookies', eval('CookiesRedisClient' + '(name="' + name + '")'))
|
||
|
|
setattr(g, name + '_account', eval('AccountRedisClient' + '(name="' + name + '")'))
|
||
|
|
return g
|
||
|
|
|
||
|
|
|
||
|
|
@app.route('/<name>/random')
|
||
|
|
def random(name):
|
||
|
|
"""
|
||
|
|
获取随机的Cookie, 访问地址如 /weibo/random
|
||
|
|
:return: 随机Cookie
|
||
|
|
"""
|
||
|
|
g = get_conn()
|
||
|
|
cookies = getattr(g, name + '_cookies').random()
|
||
|
|
return cookies
|
||
|
|
|
||
|
|
@app.route('/<name>/add/<username>/<password>')
|
||
|
|
def add(name, username, password):
|
||
|
|
"""
|
||
|
|
添加用户, 访问地址如 /weibo/add/user/password
|
||
|
|
"""
|
||
|
|
g = get_conn()
|
||
|
|
result = getattr(g, name + '_account').set(username, password)
|
||
|
|
return result
|
||
|
|
|
||
|
|
|
||
|
|
@app.route('/<name>/count')
|
||
|
|
def count(name):
|
||
|
|
"""
|
||
|
|
获取Cookies总数
|
||
|
|
"""
|
||
|
|
g = get_conn()
|
||
|
|
count = getattr(g, name + '_cookies').count()
|
||
|
|
return str(int) if isinstance(count, int) else count
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
app.run(host='0.0.0.0')
|