NTO/boilerplate
Archived
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: excel export for primitive fields

Browse Source
This commit is contained in:
Goga Coder
2025-03-11 16:10:18 +07:00
parent 5c372e8029
commit 1ed106f167
9 changed files with 149 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

123
internal/extras/excel/export.go
View File

@@ -2,7 +2,9 @@ package excel
import (
"app/internal/dialogs"
"errors"
"fmt"
"github.com/kuzgoga/fogg"
"github.com/xuri/excelize/v2"
"log/slog"
"reflect"
@@ -13,10 +15,10 @@ import (
type TableHeaders struct {
Headers []string
IgnoredFieldsIndices []int
IgnoredFieldsIndexes []int
}
func isPrimitive(valueType reflect.Type) bool {
func isPrimitiveType(valueType reflect.Type) bool {
switch valueType.Kind() {
case reflect.Bool,
reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64,
@@ -29,7 +31,20 @@ func isPrimitive(valueType reflect.Type) bool {
}
}
func ExportEntityToSpreadsheet[T any](filename, sheetName string, entity T, provider func() ([]*T, error)) error {
func DeleteDefaultSheet(file *excelize.File) error {
sheetId, err := file.GetSheetIndex("Sheet1")
if err != nil {
return err
}
if sheetId != -1 {
if err := file.DeleteSheet("Sheet1"); err != nil {
return err
}
}
return nil
}
func ExportEntitiesToSpreadsheet(filename string, exporters ...ExporterInterface) error {
file := excelize.NewFile()
defer func() {
if err := file.Close(); err != nil {
@@ -37,10 +52,27 @@ func ExportEntityToSpreadsheet[T any](filename, sheetName string, entity T, prov
}
}()
if _, err := file.NewSheet(sheetName); err != nil {
for _, exporter := range exporters {
err := ExportEntityToSpreadsheet(file, exporter.GetSheetName(), exporter.GetEntity(), exporter.GetProvider())
if err != nil {
return err
}
}
if err := DeleteDefaultSheet(file); err != nil {
return err
}
if err := file.DeleteSheet("Sheet1"); err != nil {
err := WriteData(file, filename)
if err != nil {
return err
}
return nil
}
func ExportEntityToSpreadsheet[T any](file *excelize.File, sheetName string, entity T, provider func() ([]any, error)) error {
_, err := file.NewSheet(sheetName)
if err != nil {
return err
}
@@ -59,33 +91,38 @@ func ExportEntityToSpreadsheet[T any](filename, sheetName string, entity T, prov
for i, item := range items {
structValue := reflect.ValueOf(item).Elem()
columnOffset := 0
for j := 0; j < structValue.NumField(); j++ {
if slices.Contains(headers.IgnoredFieldsIndices, j) {
if slices.Contains(headers.IgnoredFieldsIndexes, j) {
columnOffset--
continue
}
field := structValue.Field(j)
if isPrimitive(field.Type()) {
tagLiteral := string(structValue.Type().Field(j).Tag)
tag, err := fogg.Parse(tagLiteral)
if err != nil {
return err
}
if isPrimitiveType(field.Type()) {
fieldValue := field.Interface()
cellName, err := GetCellNameByIndices(j, i+1)
cellName, err := GetCellNameByIndices(j+columnOffset, i+1)
if err != nil {
return err
}
cellType := structValue.Type().Field(j).Tag.Get(CellTypeTag)
datatype := tag.GetTag("ui").GetParamOr("datatype", "")
var cellValue any
switch cellType {
case TimestampTag:
cellValue = time.Unix(field.Int(), 0)
case DurationTag:
cellValue = time.Duration(field.Int())
default:
cellValue = fieldValue
if datatype == timestampTag {
err = file.SetCellValue(sheetName, cellName, time.Unix(fieldValue.(int64), 0))
} else {
err = file.SetCellValue(sheetName, cellName, fieldValue)
}
slog.Debug("Field %s value: %v, %s\n", cellName, cellValue, cellType)
err = file.SetCellValue(sheetName, cellName, cellValue)
slog.Info(fmt.Sprintf("Field %s value: %v, %s\n", cellName, fieldValue, datatype))
if err != nil {
return err
}
@@ -93,10 +130,6 @@ func ExportEntityToSpreadsheet[T any](filename, sheetName string, entity T, prov
}
}
if err := WriteData(file, filename); err != nil {
return err
}
return nil
}
@@ -124,23 +157,43 @@ func GetCellNameByIndices(column int, row int) (string, error) {
return cellName, nil
}
func ExportHeaders(entity any) TableHeaders {
func ExportHeaders(entity any) (TableHeaders, error) {
headers := TableHeaders{}
v := reflect.TypeOf(entity)
for i := range v.NumField() {
field := v.Field(i)
displayName := field.Tag.Get("displayName")
if displayName != "" {
headers.Headers = append(headers.Headers, displayName)
tag, err := fogg.Parse(string(v.Field(i).Tag))
if err != nil {
return headers, errors.New(fmt.Sprintf("Error occured while tag parsing `%s`: %s", err, string(v.Field(i).Tag)))
}
uiTag := tag.GetTag("ui")
if uiTag == nil {
headers.IgnoredFieldsIndexes = append(headers.IgnoredFieldsIndexes, i)
continue
}
if !isPrimitiveType(v.Field(i).Type) {
headers.IgnoredFieldsIndexes = append(headers.IgnoredFieldsIndexes, i)
continue
}
label := uiTag.GetParamOr("label", uiTag.GetParamOr(excelNameTag, ""))
if label != "" {
headers.Headers = append(headers.Headers, label)
} else {
headers.IgnoredFieldsIndices = append(headers.IgnoredFieldsIndices, i)
headers.IgnoredFieldsIndexes = append(headers.IgnoredFieldsIndexes, i)
}
}
return headers
return headers, nil
}
func WriteHeaders(sheetName string, entity any, file *excelize.File) (TableHeaders, error) {
headers := ExportHeaders(entity)
headers, err := ExportHeaders(entity)
if err != nil {
return headers, err
}
for i, header := range headers.Headers {
cellName, err := GetHeaderCellNameByIndex(i)
if err != nil {
@@ -152,17 +205,19 @@ func WriteHeaders(sheetName string, entity any, file *excelize.File) (TableHeade
return headers, err
}
}
err := ApplyStyleHeaders(file, sheetName, headers)
err = ApplyStyleHeaders(file, sheetName, headers)
if err != nil {
return headers, err
}
return headers, nil
}
func GetStyleId(f *excelize.File, style *excelize.Style) (int, error) {
styleId, err := f.NewStyle(style)
if err != nil {
return 0, fmt.Errorf("ошибка при создании стиля: %w", err)
return 0, fmt.Errorf("error occured while creating a style: %w", err)
}
return styleId, nil

35
internal/extras/excel/exporter.go Normal file
View File

@@ -0,0 +1,35 @@
package excel
type ExporterInterface interface {
GetSheetName() string
GetEntity() any
GetProvider() func() ([]any, error)
}
type Exporter[T any] struct {
SheetName string
Entity T
Provider func() ([]*T, error)
}
func (e Exporter[T]) GetSheetName() string {
return e.SheetName
}
func (e Exporter[T]) GetEntity() any {
return e.Entity
}
func (e Exporter[T]) GetProvider() func() ([]any, error) {
return func() ([]any, error) {
entities, err := e.Provider()
if err != nil {
return nil, err
}
result := make([]any, len(entities))
for i, entity := range entities {
result[i] = entity
}
return result, nil
}
}

10
internal/extras/excel/tags.go
View File

@@ -1,10 +1,4 @@
package excel
const (
CellTypeTag = "cellType"
)
const (
TimestampTag = "timestamp"
DurationTag = "duration"
)
const excelNameTag string = "excel"
const timestampTag string = "datetime"

8
internal/models/models.go
View File

@@ -10,14 +10,14 @@ type PostType struct {
}
type Post struct {
Id uint `gorm:"primaryKey" ui:"hidden"`
Id uint `gorm:"primaryKey" ui:"hidden;label:\"Номер поста\""`
Text string `ui:"label:Текст"`
Deadline int64 `ui:"label:Дедлайн;datatype:datetime;"`
CreatedAt int64 `gorm:"autoCreateTime" ui:"label:Время создания; readonly; datatype:datetime;"`
CreatedAt int64 `gorm:"autoCreateTime" ui:"label:Время создания;readonly;datatype:datetime;"`
AuthorId uint `ui:"hidden" gorm:"constraint:OnUpdate:CASCADE,OnDelete:CASCADE;"`
Author Author `ui:"label:Автор; field:Name;"`
PostTypeId uint `ui:"hidden"`
PostType PostType `ui:"label:Тип; field:Name;" gorm:"constraint:OnUpdate:CASCADE,OnDelete:CASCADE;"`
PostTypeId uint `ui:"hidden; excel:Номер типа поста;"`
PostType PostType `ui:"label:Тип поста; field:Name;" gorm:"constraint:OnUpdate:CASCADE,OnDelete:CASCADE;"`
Comments []Comment `ui:"label:Комментарии; field:Text;" gorm:"many2many:comments_post;constraint:OnUpdate:CASCADE,OnDelete:CASCADE;"`
}

3
internal/services/comment.go
View File

@@ -29,6 +29,7 @@ func (service *CommentService) GetAll() ([]*Comment, error) {
comments, err := dal.Comment.Preload(field.Associations).Find()
return comments, err
}
func (service *CommentService) GetById(id uint) (*Comment, error) {
item, err := dal.Comment.Preload(field.Associations).Where(dal.Comment.Id.Eq(id)).First()
if err != nil {
@@ -40,6 +41,7 @@ func (service *CommentService) GetById(id uint) (*Comment, error) {
}
return item, nil
}
func (service *CommentService) Update(item Comment) (Comment, error) {
ReplaceEmptySlicesWithNil(&item)
err := dal.Comment.Preload(field.Associations).Save(&item)
@@ -55,6 +57,7 @@ func (service *CommentService) Update(item Comment) (Comment, error) {
return item, err
}
func (service *CommentService) Delete(id uint) error {
_, err := dal.Comment.Unscoped().Where(dal.Comment.Id.Eq(id)).Delete()
return err

9
internal/services/post.go
View File

@@ -4,7 +4,7 @@ import (
"app/internal/dal"
"app/internal/database"
"app/internal/dialogs"
excel "app/internal/extras/excel"
"app/internal/extras/excel"
"app/internal/models"
"errors"
"fmt"
@@ -64,7 +64,12 @@ func (service *PostService) Count() (int64, error) {
return amount, err
}
func (service *PostService) ExportToExcel() {
err := excel.ExportEntityToSpreadsheet("report.xlsx", "Посты", Post{}, service.GetAll)
exporter := excel.Exporter[Post]{
SheetName: "Посты",
Entity: Post{},
Provider: service.GetAll,
}
err := excel.ExportEntitiesToSpreadsheet("report.xlsx", exporter)
if err != nil {
dialogs.ErrorDialog("Ошибка экспорта", fmt.Sprintf("Ошибка при экспорте данных: %s", err))
}

11
internal/services/posttype.go
View File

@@ -7,12 +7,9 @@ import (
"app/internal/extras/excel"
"app/internal/models"
"errors"
"os/signal"
"strconv"
"syscall"
"gorm.io/gen/field"
"gorm.io/gorm"
"strconv"
)
type PostTypeService struct {
@@ -66,7 +63,6 @@ func (service *PostTypeService) Count() (int64, error) {
}
func (service *PostTypeService) ImportFromExcel() error {
signal.Ignore(syscall.SIGSEGV)
filepath, err := dialogs.OpenFileDialog("Импорт данных")
if err != nil {
return err
@@ -79,10 +75,13 @@ func (service *PostTypeService) ImportFromExcel() error {
return err
}
service.Create(PostType{
_, err = service.Create(PostType{
Id: uint(id),
Name: row[1],
})
if err != nil {
return err
}
return nil
},
})