feat: balance and reward
This commit is contained in:
		
							
								
								
									
										23
									
								
								utils/balance.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										23
									
								
								utils/balance.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,23 @@
 | 
			
		||||
import datetime
 | 
			
		||||
 | 
			
		||||
import database
 | 
			
		||||
 | 
			
		||||
import database.enums
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def add_top_up(user_id: int,
 | 
			
		||||
               amount: float,
 | 
			
		||||
               description: str,
 | 
			
		||||
               json_data: dict,
 | 
			
		||||
               commit=False):
 | 
			
		||||
    transaction = database.BalanceTransaction(
 | 
			
		||||
        user_id=user_id,
 | 
			
		||||
        type=database.enums.BalanceTransactionType.TOP_UP,
 | 
			
		||||
        amount=amount,
 | 
			
		||||
        description=description,
 | 
			
		||||
        json_data=json_data,
 | 
			
		||||
        created_at=datetime.datetime.now()
 | 
			
		||||
    )
 | 
			
		||||
    database.db.session.add(transaction)
 | 
			
		||||
    if commit:
 | 
			
		||||
        database.db.session.commit()
 | 
			
		||||
		Reference in New Issue
	
	Block a user