added manager user type
This commit is contained in:
@@ -273,11 +273,36 @@ def init_routes(main_bp):
|
||||
).group_by('extension').all()
|
||||
|
||||
# Get conversation stats
|
||||
conversation_count = Conversation.query.count()
|
||||
message_count = Message.query.count()
|
||||
attachment_count = MessageAttachment.query.count()
|
||||
conversation_total_size = db.session.query(func.sum(MessageAttachment.size)).scalar() or 0
|
||||
recent_conversations = Conversation.query.order_by(Conversation.created_at.desc()).limit(5).all()
|
||||
if current_user.is_admin:
|
||||
conversation_count = Conversation.query.count()
|
||||
message_count = Message.query.count()
|
||||
attachment_count = MessageAttachment.query.count()
|
||||
conversation_total_size = db.session.query(func.sum(MessageAttachment.size)).scalar() or 0
|
||||
recent_conversations = Conversation.query.order_by(Conversation.created_at.desc()).limit(5).all()
|
||||
else:
|
||||
# Get conversations where user is a member
|
||||
user_conversations = Conversation.query.filter(Conversation.members.any(id=current_user.id)).all()
|
||||
conversation_count = len(user_conversations)
|
||||
|
||||
# Get message count for user's conversations
|
||||
conversation_ids = [conv.id for conv in user_conversations]
|
||||
message_count = Message.query.filter(Message.conversation_id.in_(conversation_ids)).count()
|
||||
|
||||
# Get attachment count and size for user's conversations
|
||||
attachment_stats = db.session.query(
|
||||
func.count(MessageAttachment.id).label('count'),
|
||||
func.sum(MessageAttachment.size).label('total_size')
|
||||
).filter(MessageAttachment.message_id.in_(
|
||||
db.session.query(Message.id).filter(Message.conversation_id.in_(conversation_ids))
|
||||
)).first()
|
||||
|
||||
attachment_count = attachment_stats.count or 0
|
||||
conversation_total_size = attachment_stats.total_size or 0
|
||||
|
||||
# Get recent conversations for the user
|
||||
recent_conversations = Conversation.query.filter(
|
||||
Conversation.members.any(id=current_user.id)
|
||||
).order_by(Conversation.created_at.desc()).limit(5).all()
|
||||
|
||||
return render_template('dashboard/dashboard.html',
|
||||
room_count=room_count,
|
||||
|
||||
Reference in New Issue
Block a user