mirror of
https://github.com/sudoxnym/habitica.git
synced 2026-07-15 00:12:17 +00:00
Merge pull request #3960 from Alys/reportCountOwnedGear
count how many users own wondercon Unconventional Armor and Halloween Spooky Skins
This commit is contained in:
Tyler Renelle
commit
12a93c8fe3
1 changed files with 175 additions and 51 deletions
|
|
@ -1,77 +1,201 @@
|
|||
var itemsOfInterest = [
|
||||
'headAccessory_special_wondercon_red',
|
||||
'headAccessory_special_wondercon_black',
|
||||
'back_special_wondercon_black',
|
||||
'back_special_wondercon_red',
|
||||
'body_special_wondercon_red',
|
||||
'body_special_wondercon_black',
|
||||
'body_special_wondercon_gold'
|
||||
];
|
||||
var authorName = 'Alys'; // in case script author needs to know when their ...
|
||||
var authorUuid = 'd904bd62-da08-416b-a816-ba797c9ee265'; //... own data is done
|
||||
|
||||
var itemsFound = {}; // each key is one item from itemsOfInterest, and
|
||||
// the value for that key is the number of users who own it
|
||||
// ('owned' values of both true and false are counted to
|
||||
// include items lost on death)
|
||||
/**
|
||||
* database_reports/count_users_who_own_specified_gear.js
|
||||
* https://github.com/HabitRPG/habitrpg/pull/3884
|
||||
*/
|
||||
|
||||
db.users.find().forEach(function(user) {
|
||||
var owned = user.items.gear.owned;
|
||||
for (var i=0, ic=itemsOfInterest.length; i<ic; i++) {
|
||||
var itemKey = itemsOfInterest[i];
|
||||
if (itemKey in owned) {
|
||||
itemsFound[itemKey] = (itemsFound[itemKey] || 0) + 1;
|
||||
// print(user.auth.local.username + ": " + itemKey); // good for testing, bad for privacy
|
||||
}
|
||||
var thingsOfInterest = {
|
||||
'Unconventional Armor ownership': {
|
||||
'data_path': 'items.gear.owned',
|
||||
'identifyOwnershipWith': 'exists',
|
||||
'items': [
|
||||
'headAccessory_special_wondercon_red',
|
||||
'headAccessory_special_wondercon_black',
|
||||
'back_special_wondercon_black',
|
||||
'back_special_wondercon_red',
|
||||
'body_special_wondercon_red',
|
||||
'body_special_wondercon_black',
|
||||
'body_special_wondercon_gold'
|
||||
],
|
||||
},
|
||||
'Spooky Skins purchases': {
|
||||
'data_path': 'purchased.skin',
|
||||
'identifyOwnershipWith': 'true',
|
||||
'items': [
|
||||
'monster',
|
||||
'pumpkin',
|
||||
'skeleton',
|
||||
'zombie',
|
||||
'ghost',
|
||||
'shadow'
|
||||
]
|
||||
}
|
||||
};
|
||||
|
||||
var mongo = require('mongoskin');
|
||||
var _ = require('lodash');
|
||||
|
||||
//////////////// UNCOMMENT ONE OF THESE mongo.db LINES: ////////////////
|
||||
// var dbUsers = mongo.db('lefnire:mAdn3s5s@charlotte.mongohq.com:10015/habitrpg_large?auto_reconnect').collection('users'); // @lefnire production?
|
||||
// var dbUsers = mongo.db('localhost:27017/habitrpg_old?auto_reconnect').collection('users'); // @lefnire habitrpg_old
|
||||
// var dbUsers = mongo.db('localhost:27017/habitrpg?auto_reconnect').collection('users'); // for local testing by script author (e.g., vagrant install)
|
||||
if (typeof dbUsers == 'undefined') { exiting(1, 'Uncomment one of the "var dbUsers" lines!'); }
|
||||
|
||||
var thingsFound = {}; // each key is one "thing" from thingsOfInterest,
|
||||
// and the value for that key is the number of users who own it
|
||||
// (for items, 'owned' values of both true and false are counted
|
||||
// to include items lost on death)
|
||||
|
||||
var query = {}; // Not worth limiting search data with query and fields since
|
||||
var fields = {}; // this will be run over a local copy of the database?
|
||||
|
||||
console.warn('Finding data...');
|
||||
var progressCount = 1000;
|
||||
var count = 0;
|
||||
dbUsers.findEach(query, fields, {batchSize:250}, function(err, user) {
|
||||
if (err) { return exiting(1, 'ERROR! ' + err); }
|
||||
if (!user) {
|
||||
console.warn('All users found.');
|
||||
return displayData();
|
||||
}
|
||||
count++;
|
||||
|
||||
_.each(thingsOfInterest,function(obj,label){
|
||||
var data_path = obj['data_path'];
|
||||
var items = obj['items'];
|
||||
var identifyOwnershipWith = obj['identifyOwnershipWith'];
|
||||
var userOwns = path(user, data_path);
|
||||
|
||||
_.each(items,function(item){
|
||||
if ( (identifyOwnershipWith == 'exists' && item in userOwns) ||
|
||||
(identifyOwnershipWith == 'true' && userOwns[item])
|
||||
) {
|
||||
if (! thingsFound[label]) { thingsFound[label] = {}; }
|
||||
thingsFound[label][item] = (thingsFound[label][item] || 0) + 1;
|
||||
// console.warn(user.auth.local.username + ": " + label + ": " + item); // good for testing, bad for privacy
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
if (count%progressCount == 0) console.warn(count + ' ' + user._id);
|
||||
if (user._id == authorUuid) console.warn(authorName + ' processed');
|
||||
if (user._id == '9' ) console.warn('lefnire' + ' processed');
|
||||
});
|
||||
|
||||
|
||||
function displayData() {
|
||||
var today = yyyymmdd(new Date());
|
||||
var csvReport = '';
|
||||
var textReport = today + '\n';
|
||||
|
||||
_.each(thingsFound,function(obj,label){
|
||||
csvReport += '\n"' + label + '"' + '\n';
|
||||
textReport += '\n' + label + ':\n';
|
||||
var csvHeader = '"date"'; // heading row in CSV data
|
||||
var csvData = '"' + today + '"'; // data row in CSV data
|
||||
|
||||
var sortedKeys = _.sortBy(_.keys(obj), function(key){ return key; });
|
||||
_.each(sortedKeys,function(key){
|
||||
var value = obj[key];
|
||||
csvHeader += ',"' + key + '"';
|
||||
csvData += ',"' + (value || 0) + '"';
|
||||
textReport += '\t' + key + ': ' + value + '\n';
|
||||
});
|
||||
csvReport += csvHeader + '\n' + csvData + '\n';
|
||||
});
|
||||
|
||||
console.log('\nCSV DATA:\n' + csvReport + '\n\n' +
|
||||
'READABLE DATA:\n\n' + textReport + '\n\n');
|
||||
|
||||
console.warn(count + ' users searched (should be >400k)\n');
|
||||
// NB: "should be" assumes that no query filter was applied to findEach
|
||||
|
||||
return exiting(0);
|
||||
}
|
||||
|
||||
|
||||
function path(obj, path, def) {
|
||||
/**
|
||||
* Retrieve nested item from object/array
|
||||
* @param {Object|Array} obj
|
||||
* @param {String} path dot separated
|
||||
* @param {*} def default value ( if result undefined )
|
||||
* @returns {*}
|
||||
* http://stackoverflow.com/a/16190716
|
||||
* Usage: console.log(path(someObject, pathname));
|
||||
*/
|
||||
for(var i = 0,path = path.split('.'),len = path.length; i < len; i++){
|
||||
if(!obj || typeof obj !== 'object') return def;
|
||||
obj = obj[path[i]];
|
||||
}
|
||||
if(obj === 'undefined') return def;
|
||||
return obj;
|
||||
}
|
||||
|
||||
|
||||
function yyyymmdd(date) {
|
||||
var yyyy = date.getFullYear().toString();
|
||||
var mm = (date.getMonth()+1).toString();
|
||||
var dd = date.getDate().toString();
|
||||
return yyyy + "-" + (mm[1]?mm:"0"+mm[0]) + "-" + (dd[1]?dd:"0"+dd[0]);
|
||||
}
|
||||
var today = yyyymmdd(new Date());
|
||||
|
||||
var header = '"date"'; // heading row in CSV data
|
||||
var data = '"' + today + '"'; // data row in CSV data
|
||||
for (var i=0, ic=itemsOfInterest.length; i<ic; i++) {
|
||||
var itemKey = itemsOfInterest[i];
|
||||
header += ',"' + itemKey + '"';
|
||||
data += ',"' + (itemsFound[itemKey] || 0) + '"';
|
||||
|
||||
function exiting(code, msg) {
|
||||
code = code || 0; // 0 = success
|
||||
if (code && !msg) { msg = 'ERROR!'; }
|
||||
if (msg) {
|
||||
if (code) { console.error(msg); }
|
||||
else { console.log( msg); }
|
||||
}
|
||||
process.exit(code);
|
||||
}
|
||||
|
||||
print("\nCSV DATA:\n");
|
||||
print(header);
|
||||
print(data);
|
||||
print("\nREADABLE DATA:\n");
|
||||
print(today);
|
||||
print(JSON.stringify(itemsFound, null, " "));
|
||||
print("\n");
|
||||
|
||||
/* SAMPLE OUTPUT (STDOUT and STDERR):
|
||||
$ node database_reports/count_users_who_own_specified_gear.js
|
||||
|
||||
/* SAMPLE OUTPUT:
|
||||
> load("count_users_who_own_specified_gear.js")
|
||||
Finding data...
|
||||
Alys processed
|
||||
lefnire processed
|
||||
All users found.
|
||||
|
||||
CSV DATA:
|
||||
|
||||
"date","headAccessory_special_wondercon_red","headAccessory_special_wondercon_black","back_special_wondercon_black","back_special_wondercon_red","body_special_wondercon_red","body_special_wondercon_black","body_special_wondercon_gold"
|
||||
"2014-08-17","5","4","3","3","3","3","3"
|
||||
"Unconventional Armor ownership"
|
||||
"date","back_special_wondercon_black","back_special_wondercon_red","body_special_wondercon_black","body_special_wondercon_gold","body_special_wondercon_red","headAccessory_special_wondercon_black","headAccessory_special_wondercon_red"
|
||||
"2014-09-01","7","7","7","7","7","7","9"
|
||||
|
||||
"Spooky Skins purchases"
|
||||
"date","ghost","monster","pumpkin","shadow","skeleton","zombie"
|
||||
"2014-09-01","2","3","3","6","4","3"
|
||||
|
||||
|
||||
READABLE DATA:
|
||||
|
||||
2014-08-17
|
||||
{
|
||||
"headAccessory_special_wondercon_red": 5,
|
||||
"headAccessory_special_wondercon_black": 4,
|
||||
"back_special_wondercon_black": 3,
|
||||
"back_special_wondercon_red": 3,
|
||||
"body_special_wondercon_red": 3,
|
||||
"body_special_wondercon_black": 3,
|
||||
"body_special_wondercon_gold": 3
|
||||
}
|
||||
2014-09-01
|
||||
|
||||
Unconventional Armor ownership:
|
||||
back_special_wondercon_black: 7
|
||||
back_special_wondercon_red: 7
|
||||
body_special_wondercon_black: 7
|
||||
body_special_wondercon_gold: 7
|
||||
body_special_wondercon_red: 7
|
||||
headAccessory_special_wondercon_black: 7
|
||||
headAccessory_special_wondercon_red: 9
|
||||
|
||||
Spooky Skins purchases:
|
||||
ghost: 2
|
||||
monster: 3
|
||||
pumpkin: 3
|
||||
shadow: 6
|
||||
skeleton: 4
|
||||
zombie: 3
|
||||
|
||||
|
||||
true
|
||||
>
|
||||
|
||||
400100 users searched (should be >400k)
|
||||
|
||||
*/
|
||||
|
|
|
|||
Loading…
Reference in a new issue