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.
59 lines
1.3 KiB
59 lines
1.3 KiB
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')
|
|
|