show transaction pictures in data management page

pkg/api/data_managements.go

@@ -26,6 +26,7 @@ type DataManagementsApi struct {
 	users                    *services.UserService
 	accounts                 *services.AccountService
 	transactions             *services.TransactionService
+	pictures                 *services.TransactionPictureService
 	categories               *services.TransactionCategoryService
 	tags                     *services.TransactionTagService
 	templates                *services.TransactionTemplateService
@@ -43,6 +44,7 @@ var (
 		users:                    services.Users,
 		accounts:                 services.Accounts,
 		transactions:             services.Transactions,
+		pictures:                 services.TransactionPictures,
 		categories:               services.TransactionCategories,
 		tags:                     services.TransactionTags,
 		templates:                services.TransactionTemplates,
@@ -90,6 +92,13 @@ func (a *DataManagementsApi) DataStatisticsHandler(c *core.WebContext) (any, *er
 		return nil, errs.ErrOperationFailed
 	}
 
+	totalTransactionPictureCount, err := a.pictures.GetTotalTransactionPicturesCountByUid(c, uid)
+
+	if err != nil {
+		log.Errorf(c, "[data_managements.DataStatisticsHandler] failed to get total transaction picture count for user \"uid:%d\", because %s", uid, err.Error())
+		return nil, errs.ErrOperationFailed
+	}
+
 	totalTransactionTemplateCount, err := a.templates.GetTotalNormalTemplateCountByUid(c, uid)
 
 	if err != nil {
@@ -109,6 +118,7 @@ func (a *DataManagementsApi) DataStatisticsHandler(c *core.WebContext) (any, *er
 		TotalTransactionCategoryCount:  totalTransactionCategoryCount,
 		TotalTransactionTagCount:       totalTransactionTagCount,
 		TotalTransactionCount:          totalTransactionCount,
+		TotalTransactionPictureCount:   totalTransactionPictureCount,
 		TotalTransactionTemplateCount:  totalTransactionTemplateCount,
 		TotalScheduledTransactionCount: totalScheduledTransactionCount,
 	}
@@ -148,6 +158,13 @@ func (a *DataManagementsApi) ClearDataHandler(c *core.WebContext) (any, *errs.Er
 		return nil, errs.Or(err, errs.ErrOperationFailed)
 	}
 
+	err = a.pictures.DeleteAllPictures(c, uid)
+
+	if err != nil {
+		log.Errorf(c, "[data_managements.ClearDataHandler] failed to delete all transaction pictures, because %s", err.Error())
+		return nil, errs.Or(err, errs.ErrOperationFailed)
+	}
+
 	err = a.transactions.DeleteAllTransactions(c, uid)
 
 	if err != nil {

pkg/models/data_management.go

@@ -11,6 +11,7 @@ type DataStatisticsResponse struct {
 	TotalTransactionCategoryCount  int64 `json:"totalTransactionCategoryCount,string"`
 	TotalTransactionTagCount       int64 `json:"totalTransactionTagCount,string"`
 	TotalTransactionCount          int64 `json:"totalTransactionCount,string"`
+	TotalTransactionPictureCount   int64 `json:"totalTransactionPictureCount,string"`
 	TotalTransactionTemplateCount  int64 `json:"totalTransactionTemplateCount,string"`
 	TotalScheduledTransactionCount int64 `json:"totalScheduledTransactionCount,string"`
 }

pkg/services/transaction_pictures.go

@@ -38,6 +38,17 @@ var (
 	}
 )
 
+// GetTotalTransactionPicturesCountByUid returns total transaction pictures count of user
+func (s *TransactionPictureService) GetTotalTransactionPicturesCountByUid(c core.Context, uid int64) (int64, error) {
+	if uid <= 0 {
+		return 0, errs.ErrUserIdInvalid
+	}
+
+	count, err := s.UserDataDB(uid).NewSession(c).Where("uid=? AND deleted=?", uid, false).Count(&models.TransactionPictureInfo{})
+
+	return count, err
+}
+
 // GetPictureInfoByPictureId returns a transaction picture info model according to transaction picture id
 func (s *TransactionPictureService) GetPictureInfoByPictureId(c core.Context, uid int64, pictureId int64) (*models.TransactionPictureInfo, error) {
 	if uid <= 0 {
@@ -138,3 +149,27 @@ func (s *TransactionPictureService) UploadPicture(c core.Context, pictureInfo *m
 		return err
 	})
 }
+
+// DeleteAllPictures deletes all existed transaction pictures from database
+func (s *TransactionPictureService) DeleteAllPictures(c core.Context, uid int64) error {
+	if uid <= 0 {
+		return errs.ErrUserIdInvalid
+	}
+
+	now := time.Now().Unix()
+
+	updateModel := &models.TransactionPictureInfo{
+		Deleted:         true,
+		DeletedUnixTime: now,
+	}
+
+	return s.UserDataDB(uid).DoTransaction(c, func(sess *xorm.Session) error {
+		_, err := sess.Cols("deleted", "deleted_unix_time").Where("uid=? AND deleted=?", uid, false).Update(updateModel)
+
+		if err != nil {
+			return err
+		}
+
+		return nil
+	})
+}