Restructure stats command
This commit is contained in:
Florian Baumann
parent
1ce3920c03
commit
3691890b95
@ -16,197 +16,231 @@ func init() {
|
||||
|
||||
var statsCmd = &cobra.Command{
|
||||
Aliases: []string{"stats"},
|
||||
Use: "stats <prefix> <n>",
|
||||
Use: "stats",
|
||||
Short: "Shows statistics of the collection",
|
||||
SilenceErrors: true,
|
||||
RunE: func(cmd *cobra.Command, args []string) error {
|
||||
|
||||
client := storageConnect()
|
||||
coll := &Collection{client.Database("serra").Collection("cards")}
|
||||
totalcoll := &Collection{client.Database("serra").Collection("total")}
|
||||
l := Logger()
|
||||
defer storageDisconnect(client)
|
||||
|
||||
// Value and Card Numbers
|
||||
stats, _ := coll.storageAggregate(mongo.Pipeline{
|
||||
bson.D{
|
||||
{"$group", bson.D{
|
||||
{"_id", nil},
|
||||
{"value", bson.D{{"$sum", bson.D{{"$multiply", bson.A{getCurrencyField(false), "$serra_count"}}}}}},
|
||||
{"value_foil", bson.D{{"$sum", bson.D{{"$multiply", bson.A{getCurrencyField(true), "$serra_count_foil"}}}}}},
|
||||
{"count", bson.D{{"$sum", bson.D{{"$multiply", bson.A{1.0, "$serra_count"}}}}}},
|
||||
{"count_foil", bson.D{{"$sum", "$serra_count_foil"}}},
|
||||
{"rarity", bson.D{{"$sum", "$rarity"}}},
|
||||
{"unique", bson.D{{"$sum", 1}}},
|
||||
}},
|
||||
},
|
||||
bson.D{
|
||||
{"$addFields", bson.D{
|
||||
{"count_all", bson.D{{"$sum", bson.A{"$count", "$count_foil"}}}},
|
||||
}},
|
||||
},
|
||||
})
|
||||
fmt.Printf("%sCards %s\n", Green, Reset)
|
||||
fmt.Printf("Total: %s%.0f%s\n", Yellow, stats[0]["count_all"], Reset)
|
||||
fmt.Printf("Unique: %s%d%s\n", Purple, stats[0]["unique"], Reset)
|
||||
fmt.Printf("Normal: %s%.0f%s\n", Purple, stats[0]["count"], Reset)
|
||||
fmt.Printf("Foil: %s%d%s\n", Purple, stats[0]["count_foil"], Reset)
|
||||
|
||||
reserved, _ := coll.storageAggregate(mongo.Pipeline{
|
||||
bson.D{
|
||||
{"$match", bson.D{
|
||||
{"reserved", true}}}},
|
||||
bson.D{
|
||||
{"$group", bson.D{
|
||||
{"_id", nil},
|
||||
{"count", bson.D{{"$sum", 1}}},
|
||||
}}},
|
||||
})
|
||||
|
||||
var count_reserved int32
|
||||
if len(reserved) > 0 {
|
||||
count_reserved = reserved[0]["count"].(int32)
|
||||
}
|
||||
fmt.Printf("Reserved List: %s%d%s\n", Yellow, count_reserved, Reset)
|
||||
|
||||
// Rarities
|
||||
rar, _ := coll.storageAggregate(mongo.Pipeline{
|
||||
bson.D{
|
||||
{"$group", bson.D{
|
||||
{"_id", "$rarity"},
|
||||
{"count", bson.D{{"$sum", bson.D{{"$multiply", bson.A{1.0, "$serra_count"}}}}}},
|
||||
}}},
|
||||
bson.D{
|
||||
{"$sort", bson.D{
|
||||
{"_id", 1},
|
||||
}}},
|
||||
})
|
||||
ri := convertRarities(rar)
|
||||
fmt.Printf("\n%sRarity%s\n", Green, Reset)
|
||||
fmt.Printf("Mythics: %s%.0f%s\n", Pink, ri.Mythics, Reset)
|
||||
fmt.Printf("Rares: %s%.0f%s\n", Pink, ri.Rares, Reset)
|
||||
fmt.Printf("Uncommons: %s%.0f%s\n", Yellow, ri.Uncommons, Reset)
|
||||
fmt.Printf("Commons: %s%.0f%s\n", Purple, ri.Commons, Reset)
|
||||
|
||||
// Colors
|
||||
sets, _ := coll.storageAggregate(mongo.Pipeline{
|
||||
bson.D{
|
||||
{"$match", bson.D{
|
||||
{"coloridentity", bson.D{{"$size", 1}}}}}},
|
||||
bson.D{
|
||||
{"$group", bson.D{
|
||||
{"_id", "$coloridentity"},
|
||||
{"count", bson.D{{"$sum", bson.D{{"$multiply", bson.A{1.0, "$serra_count"}}}}}},
|
||||
}}},
|
||||
bson.D{
|
||||
{"$sort", bson.D{
|
||||
{"count", -1},
|
||||
}}},
|
||||
})
|
||||
|
||||
fmt.Printf("\n%sColors%s\n", Green, Reset)
|
||||
for _, set := range sets {
|
||||
x, _ := set["_id"].(primitive.A)
|
||||
s := []interface{}(x)
|
||||
fmt.Printf("%s: %s%.0f%s\n", convertManaSymbols(s), Purple, set["count"], Reset)
|
||||
}
|
||||
// Artists
|
||||
artists, _ := coll.storageAggregate(mongo.Pipeline{
|
||||
bson.D{
|
||||
{"$group", bson.D{
|
||||
{"_id", "$artist"},
|
||||
{"count", bson.D{{"$sum", 1}}},
|
||||
}}},
|
||||
bson.D{
|
||||
{"$sort", bson.D{
|
||||
{"count", -1},
|
||||
}}},
|
||||
bson.D{
|
||||
{"$limit", 10}},
|
||||
})
|
||||
fmt.Printf("\n%sTop Artists%s\n", Green, Reset)
|
||||
for _, artist := range artists {
|
||||
fmt.Printf("%s: %s%d%s\n", artist["_id"].(string), Purple, artist["count"], Reset)
|
||||
}
|
||||
|
||||
// Mana Curve of Collection
|
||||
cmc, _ := coll.storageAggregate(mongo.Pipeline{
|
||||
bson.D{
|
||||
{"$group", bson.D{
|
||||
{"_id", "$cmc"},
|
||||
// {"count", bson.D{{"$sum", bson.D{{"$multiply", bson.A{1.0, "$serra_count"}}}}}},
|
||||
{"count", bson.D{{"$sum", 1}}},
|
||||
}}},
|
||||
bson.D{
|
||||
{"$sort", bson.D{
|
||||
{"_id", 1},
|
||||
}}},
|
||||
})
|
||||
fmt.Printf("\n%sMana Curve%s\n", Green, Reset)
|
||||
for _, mc := range cmc {
|
||||
fmt.Printf("%.0f: %s%d%s\n", mc["_id"], Purple, mc["count"], Reset)
|
||||
}
|
||||
|
||||
// Show cards added per month
|
||||
fmt.Printf("\n%sCards added over time%s\n", Green, Reset)
|
||||
type Caot struct {
|
||||
Id struct {
|
||||
Year int32 `mapstructure:"year"`
|
||||
Month int32 `mapstructure:"month"`
|
||||
} `mapstructure:"_id"`
|
||||
Count int32 `mapstructure:"count"`
|
||||
}
|
||||
caot, _ := coll.storageAggregate(mongo.Pipeline{
|
||||
bson.D{
|
||||
{"$project", bson.D{
|
||||
{"month", bson.D{
|
||||
{"$month", "$serra_created"}}},
|
||||
{"year", bson.D{
|
||||
{"$year", "$serra_created"}},
|
||||
}},
|
||||
}},
|
||||
bson.D{
|
||||
{"$group", bson.D{
|
||||
{"_id", bson.D{{"month", "$month"}, {"year", "$year"}}},
|
||||
{"count", bson.D{{"$sum", 1}}},
|
||||
}},
|
||||
},
|
||||
bson.D{
|
||||
{"$sort", bson.D{{"_id.year", 1}, {"_id.month", 1}}},
|
||||
},
|
||||
})
|
||||
for _, mo := range caot {
|
||||
moo := new(Caot)
|
||||
mapstructure.Decode(mo, moo)
|
||||
fmt.Printf("%d-%02d: %s%d%s\n", moo.Id.Year, moo.Id.Month, Purple, moo.Count, Reset)
|
||||
}
|
||||
|
||||
// Total Value
|
||||
fmt.Printf("\n%sTotal Value%s\n", Green, Reset)
|
||||
normalValue, err := getFloat64(stats[0]["value"])
|
||||
if err != nil {
|
||||
l.Error(err)
|
||||
normalValue = 0
|
||||
}
|
||||
foilValue, err := getFloat64(stats[0]["value_foil"])
|
||||
if err != nil {
|
||||
l.Error(err)
|
||||
foilValue = 0
|
||||
}
|
||||
count_all, err := getFloat64(stats[0]["count_all"])
|
||||
if err != nil {
|
||||
l.Error(err)
|
||||
foilValue = 0
|
||||
}
|
||||
totalValue := normalValue + foilValue
|
||||
fmt.Printf("Total: %s%.2f%s%s\n", Pink, totalValue, getCurrency(), Reset)
|
||||
fmt.Printf("Normal: %s%.2f%s%s\n", Pink, normalValue, getCurrency(), Reset)
|
||||
fmt.Printf("Foils: %s%.2f%s%s\n", Pink, foilValue, getCurrency(), Reset)
|
||||
fmt.Printf("Average Card: %s%.2f%s%s\n", Pink, totalValue/count_all, getCurrency(), Reset)
|
||||
total, _ := totalcoll.storageFindTotal()
|
||||
|
||||
fmt.Printf("History: \n")
|
||||
showPriceHistory(total.Value, "* ", true)
|
||||
Stats()
|
||||
return nil
|
||||
},
|
||||
}
|
||||
|
||||
func Stats() {
|
||||
client := storageConnect()
|
||||
coll := &Collection{client.Database("serra").Collection("cards")}
|
||||
totalcoll := &Collection{client.Database("serra").Collection("total")}
|
||||
l := Logger()
|
||||
defer storageDisconnect(client)
|
||||
|
||||
// Show Value Stats
|
||||
showValueStats(coll, totalcoll)
|
||||
|
||||
// Reserved List
|
||||
showReservedListStats(coll)
|
||||
|
||||
// Rarities
|
||||
showRarityStats(coll)
|
||||
|
||||
// Colors
|
||||
showColorStats(coll)
|
||||
|
||||
// Artists
|
||||
showArtistStats(coll)
|
||||
|
||||
// Mana Curve of Collection
|
||||
showManaCurveStats(coll)
|
||||
|
||||
// Show cards added per month
|
||||
showCardsAddedPerMonth(coll)
|
||||
|
||||
}
|
||||
|
||||
func showValueStats(coll *Collection, totalcoll *Collection) {
|
||||
// Value and Card Numbers
|
||||
stats, _ := coll.storageAggregate(mongo.Pipeline{
|
||||
bson.D{
|
||||
{"$group", bson.D{
|
||||
{"_id", nil},
|
||||
{"value", bson.D{{"$sum", bson.D{{"$multiply", bson.A{getCurrencyField(false), "$serra_count"}}}}}},
|
||||
{"value_foil", bson.D{{"$sum", bson.D{{"$multiply", bson.A{getCurrencyField(true), "$serra_count_foil"}}}}}},
|
||||
{"count", bson.D{{"$sum", bson.D{{"$multiply", bson.A{1.0, "$serra_count"}}}}}},
|
||||
{"count_foil", bson.D{{"$sum", "$serra_count_foil"}}},
|
||||
{"rarity", bson.D{{"$sum", "$rarity"}}},
|
||||
{"unique", bson.D{{"$sum", 1}}},
|
||||
}},
|
||||
},
|
||||
bson.D{
|
||||
{"$addFields", bson.D{
|
||||
{"count_all", bson.D{{"$sum", bson.A{"$count", "$count_foil"}}}},
|
||||
}},
|
||||
},
|
||||
})
|
||||
fmt.Printf("%sCards %s\n", Green, Reset)
|
||||
fmt.Printf("Total: %s%.0f%s\n", Yellow, stats[0]["count_all"], Reset)
|
||||
fmt.Printf("Unique: %s%d%s\n", Purple, stats[0]["unique"], Reset)
|
||||
fmt.Printf("Normal: %s%.0f%s\n", Purple, stats[0]["count"], Reset)
|
||||
fmt.Printf("Foil: %s%d%s\n", Purple, stats[0]["count_foil"], Reset)
|
||||
|
||||
// Total Value
|
||||
fmt.Printf("\n%sTotal Value%s\n", Green, Reset)
|
||||
normalValue, err := getFloat64(stats[0]["value"])
|
||||
if err != nil {
|
||||
l.Error(err)
|
||||
normalValue = 0
|
||||
}
|
||||
foilValue, err := getFloat64(stats[0]["value_foil"])
|
||||
if err != nil {
|
||||
l.Error(err)
|
||||
foilValue = 0
|
||||
}
|
||||
count_all, err := getFloat64(stats[0]["count_all"])
|
||||
if err != nil {
|
||||
l.Error(err)
|
||||
foilValue = 0
|
||||
}
|
||||
totalValue := normalValue + foilValue
|
||||
fmt.Printf("Total: %s%.2f%s%s\n", Pink, totalValue, getCurrency(), Reset)
|
||||
fmt.Printf("Normal: %s%.2f%s%s\n", Pink, normalValue, getCurrency(), Reset)
|
||||
fmt.Printf("Foils: %s%.2f%s%s\n", Pink, foilValue, getCurrency(), Reset)
|
||||
fmt.Printf("Average Card: %s%.2f%s%s\n", Pink, totalValue/count_all, getCurrency(), Reset)
|
||||
total, _ := totalcoll.storageFindTotal()
|
||||
|
||||
fmt.Printf("History: \n")
|
||||
showPriceHistory(total.Value, "* ", true)
|
||||
}
|
||||
|
||||
func showReservedListStats(coll *Collection) {
|
||||
reserved, _ := coll.storageAggregate(mongo.Pipeline{
|
||||
bson.D{
|
||||
{"$match", bson.D{
|
||||
{"reserved", true}}}},
|
||||
bson.D{
|
||||
{"$group", bson.D{
|
||||
{"_id", nil},
|
||||
{"count", bson.D{{"$sum", 1}}},
|
||||
}}},
|
||||
})
|
||||
|
||||
var count_reserved int32
|
||||
if len(reserved) > 0 {
|
||||
count_reserved = reserved[0]["count"].(int32)
|
||||
}
|
||||
fmt.Printf("Reserved List: %s%d%s\n", Yellow, count_reserved, Reset)
|
||||
}
|
||||
|
||||
func showRarityStats(coll *Collection) {
|
||||
rar, _ := coll.storageAggregate(mongo.Pipeline{
|
||||
bson.D{
|
||||
{"$group", bson.D{
|
||||
{"_id", "$rarity"},
|
||||
{"count", bson.D{{"$sum", bson.D{{"$multiply", bson.A{1.0, "$serra_count"}}}}}},
|
||||
}}},
|
||||
bson.D{
|
||||
{"$sort", bson.D{
|
||||
{"_id", 1},
|
||||
}}},
|
||||
})
|
||||
ri := convertRarities(rar)
|
||||
fmt.Printf("\n%sRarity%s\n", Green, Reset)
|
||||
fmt.Printf("Mythics: %s%.0f%s\n", Pink, ri.Mythics, Reset)
|
||||
fmt.Printf("Rares: %s%.0f%s\n", Pink, ri.Rares, Reset)
|
||||
fmt.Printf("Uncommons: %s%.0f%s\n", Yellow, ri.Uncommons, Reset)
|
||||
fmt.Printf("Commons: %s%.0f%s\n", Purple, ri.Commons, Reset)
|
||||
}
|
||||
|
||||
func showCardsAddedPerMonth(coll *Collection) {
|
||||
fmt.Printf("\n%sCards added over time%s\n", Green, Reset)
|
||||
type Caot struct {
|
||||
Id struct {
|
||||
Year int32 `mapstructure:"year"`
|
||||
Month int32 `mapstructure:"month"`
|
||||
} `mapstructure:"_id"`
|
||||
Count int32 `mapstructure:"count"`
|
||||
}
|
||||
caot, _ := coll.storageAggregate(mongo.Pipeline{
|
||||
bson.D{
|
||||
{"$project", bson.D{
|
||||
{"month", bson.D{
|
||||
{"$month", "$serra_created"}}},
|
||||
{"year", bson.D{
|
||||
{"$year", "$serra_created"}},
|
||||
}},
|
||||
}},
|
||||
bson.D{
|
||||
{"$group", bson.D{
|
||||
{"_id", bson.D{{"month", "$month"}, {"year", "$year"}}},
|
||||
{"count", bson.D{{"$sum", 1}}},
|
||||
}},
|
||||
},
|
||||
bson.D{
|
||||
{"$sort", bson.D{{"_id.year", 1}, {"_id.month", 1}}},
|
||||
},
|
||||
})
|
||||
for _, mo := range caot {
|
||||
moo := new(Caot)
|
||||
mapstructure.Decode(mo, moo)
|
||||
fmt.Printf("%d-%02d: %s%d%s\n", moo.Id.Year, moo.Id.Month, Purple, moo.Count, Reset)
|
||||
}
|
||||
}
|
||||
|
||||
func showManaCurveStats(coll *Collection) {
|
||||
cmc, _ := coll.storageAggregate(mongo.Pipeline{
|
||||
bson.D{
|
||||
{"$group", bson.D{
|
||||
{"_id", "$cmc"},
|
||||
{"count", bson.D{{"$sum", 1}}},
|
||||
}}},
|
||||
bson.D{
|
||||
{"$sort", bson.D{
|
||||
{"_id", 1},
|
||||
}}},
|
||||
})
|
||||
fmt.Printf("\n%sMana Curve%s\n", Green, Reset)
|
||||
for _, mc := range cmc {
|
||||
fmt.Printf("%.0f: %s%d%s\n", mc["_id"], Purple, mc["count"], Reset)
|
||||
}
|
||||
}
|
||||
|
||||
func showArtistStats(coll *Collection) {
|
||||
artists, _ := coll.storageAggregate(mongo.Pipeline{
|
||||
bson.D{
|
||||
{"$group", bson.D{
|
||||
{"_id", "$artist"},
|
||||
{"count", bson.D{{"$sum", 1}}},
|
||||
}}},
|
||||
bson.D{
|
||||
{"$sort", bson.D{
|
||||
{"count", -1},
|
||||
}}},
|
||||
bson.D{
|
||||
{"$limit", 10}},
|
||||
})
|
||||
fmt.Printf("\n%sTop Artists%s\n", Green, Reset)
|
||||
for _, artist := range artists {
|
||||
fmt.Printf("%s: %s%d%s\n", artist["_id"].(string), Purple, artist["count"], Reset)
|
||||
}
|
||||
}
|
||||
|
||||
func showColorStats(coll *Collection) {
|
||||
sets, _ := coll.storageAggregate(mongo.Pipeline{
|
||||
bson.D{
|
||||
{"$match", bson.D{
|
||||
{"coloridentity", bson.D{{"$size", 1}}}}}},
|
||||
bson.D{
|
||||
{"$group", bson.D{
|
||||
{"_id", "$coloridentity"},
|
||||
{"count", bson.D{{"$sum", bson.D{{"$multiply", bson.A{1.0, "$serra_count"}}}}}},
|
||||
}}},
|
||||
bson.D{
|
||||
{"$sort", bson.D{
|
||||
{"count", -1},
|
||||
}}},
|
||||
})
|
||||
|
||||
fmt.Printf("\n%sColors%s\n", Green, Reset)
|
||||
for _, set := range sets {
|
||||
x, _ := set["_id"].(primitive.A)
|
||||
s := []interface{}(x)
|
||||
fmt.Printf("%s: %s%.0f%s\n", convertManaSymbols(s), Purple, set["count"], Reset)
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user