def create(cls, document):
document['_id'] = ObjectId()
trigger_tasks.apply_async(('map_created', {'map': document}))
return super().create(document)
def update(cls, _id, document, other_operators={}, projection={}):
document = super().update(_id, document, other_operators,
projection)
trigger_tasks.apply_async(('user_updated', {'user': document}))
return document
def create(cls, document):
# document['order_count'] = 0
# document['subscription_order_count'] = 0
# document['referral_id'] = ''.join(random.choice(string.ascii_lowercase + string.digits) for _ in range(7))
# document['referral_count'] = 0
# document['referral_order_count'] = 0
# document['referral_subscription_count'] = 0
# document['referral_subscription_order_count'] = 0
document['_id'] = ObjectId()
document['user_id'] = document['_id']
if 'password' not in document:
document['password'] = ''.join(
random.choice(string.ascii_lowercase + string.digits)
for _ in range(9))
document['password'] = '******'.join([
document['password'][:3], document['password'][3:6],
document['password'][6:]
])
has_generated_password = True
else:
has_generated_password = False
trigger_tasks.apply_async(('user_created', {
'user':
document,
'has_generated_password':
has_generated_password
}))
return super().create(document)