Add reconcile_subscribers function
For every user, check their subscription status with the store.
This commit is contained in:
parent
a5b2259b94
commit
d7ad60fb9d
34
manage.py
34
manage.py
@ -1,7 +1,41 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
from __future__ import print_function
|
||||
|
||||
import logging
|
||||
from flask import current_app
|
||||
from pillar import cli
|
||||
from pillar.cli import manager
|
||||
from cloud import app
|
||||
|
||||
log = logging.getLogger(__name__)
|
||||
|
||||
|
||||
@manager.command
|
||||
def reconcile_subscribers():
|
||||
"""For every user, check their subscription status with the store."""
|
||||
from pillar.auth.subscriptions import fetch_user
|
||||
|
||||
users_coll = current_app.data.driver.db['users']
|
||||
unsubscribed_users = []
|
||||
for user in users_coll.find():
|
||||
print('Processing %s' % user['email'])
|
||||
if 'roles' in user and 'subscriber' in user['roles']:
|
||||
print('Checking subscription')
|
||||
user_store = fetch_user(user['email'])
|
||||
if user_store['cloud_access'] == 0:
|
||||
print('Removing subscriber role')
|
||||
users_coll.update(
|
||||
{'_id': user['_id']},
|
||||
{'$pull': {'roles': 'subscriber'}})
|
||||
unsubscribed_users.append(user['email'])
|
||||
|
||||
if not unsubscribed_users:
|
||||
return
|
||||
|
||||
print('The following users have been unsubscribed')
|
||||
for user in unsubscribed_users:
|
||||
print(user)
|
||||
|
||||
cli.manager.app = app
|
||||
cli.manager.run()
|
||||
|
Loading…
x
Reference in New Issue
Block a user