更新config.js 可自訂sa路徑

This commit is contained in:
liaojack8 2020-07-05 22:15:21 +08:00
parent cbcdeb52e8
commit a12656ebad
3 changed files with 36 additions and 3 deletions

View File

@ -20,5 +20,7 @@ const AUTH = { // 如果您拥有service account的json授权文件可将其
tg_token: 'bot_token', // 你的 telegram robot 的 token获取方法参见 https://core.telegram.org/bots#6-botfather tg_token: 'bot_token', // 你的 telegram robot 的 token获取方法参见 https://core.telegram.org/bots#6-botfather
tg_whitelist: ['your_tg_username'] // 你的tg username(t.me/username)bot只会执行这个列表里的用户所发送的指令 tg_whitelist: ['your_tg_username'] // 你的tg username(t.me/username)bot只会执行这个列表里的用户所发送的指令
} }
//-----------------------MOD-----------------------
const SA_Path = '../sa' //sa路徑配置, 給定絕對路徑或是以src為當前路徑給定相對路徑
module.exports = { AUTH, PARALLEL_LIMIT, RETRY_LIMIT, TIMEOUT_BASE, TIMEOUT_MAX, LOG_DELAY, PAGE_SIZE, DEFAULT_TARGET } module.exports = { AUTH, PARALLEL_LIMIT, RETRY_LIMIT, TIMEOUT_BASE, TIMEOUT_MAX, LOG_DELAY, PAGE_SIZE, DEFAULT_TARGET, SA_Path }

31
count2 Normal file
View File

@ -0,0 +1,31 @@
#!/usr/bin/env node
const { argv } = require('yargs')
.usage('用法: ./$0 <目录ID> [options]')
.example('./$0 1ULY8ISgWSOVc0UrzejykVgXfVL_I4r75', '获取 https://drive.google.com/drive/folders/1ULY8ISgWSOVc0UrzejykVgXfVL_I4r75 内包含的的所有文件的统计信息')
.example('./$0 root -s size -t html -o out.html', '获取个人盘根目录统计信息结果以HTML表格输出根据总大小逆序排列保存到本目录下的out.html文件中不存在则新建存在则覆盖')
.example('./$0 root -s name -t json -o out.json', '获取个人盘根目录统计信息结果以JSON格式输出根据文件扩展名排序保存到本目录下的out.json文件中')
.example('./$0 root -t all -o all.json', '获取个人盘根目录统计信息将所有文件信息包括文件夹以JSON格式输出保存到本目录下的all.json文件中')
.alias('u', 'update')
.describe('u', '强制从线上获取信息(无视是否存在本地缓存)')
.alias('N', 'not_teamdrive')
.describe('N', '如果不是团队盘链接,可以加上此参数以提高接口查询效率,降低延迟。如果要统计的是个人盘且./sa目录下的service account没有相关权限请确保加上此参数以使用个人的auth信息进行查询')
.alias('S', 'service_account')
.describe('S', '指定使用service account进行统计前提是必须在sa目录下放置SA json文件')
.alias('s', 'sort')
.describe('s', '统计结果排序方法,可选值 name 或 size不填则默认根据文件数量逆序排列')
.alias('t', 'type')
.describe('t', '统计结果输出类型,可选值 html/json/allall表示输出所有文件json数据最好搭配 -o 使用。不填则默认输出命令行表格')
.alias('o', 'output')
.describe('o', '统计结果输出文件,适合搭配 -t 使用')
.help('h')
.alias('h', 'help')
const { count, validate_fid } = require('./src/gd')
const [fid] = argv._
if (validate_fid(fid)) {
const { update, sort, type, output, not_teamdrive, service_account } = argv
count({ fid, update, sort, type, output, not_teamdrive, service_account }).catch(console.error)
} else {
console.warn('目录ID缺失或格式错误')
}

View File

@ -8,7 +8,7 @@ const HttpsProxyAgent = require('https-proxy-agent')
const { GoogleToken } = require('gtoken') const { GoogleToken } = require('gtoken')
const handle_exit = require('signal-exit') const handle_exit = require('signal-exit')
const { AUTH, RETRY_LIMIT, PARALLEL_LIMIT, TIMEOUT_BASE, TIMEOUT_MAX, LOG_DELAY, PAGE_SIZE, DEFAULT_TARGET } = require('../config') const { AUTH, RETRY_LIMIT, PARALLEL_LIMIT, TIMEOUT_BASE, TIMEOUT_MAX, LOG_DELAY, PAGE_SIZE, DEFAULT_TARGET, SA_Path } = require('../config')
const { db } = require('../db') const { db } = require('../db')
const { make_table, make_tg_table, make_html, summary } = require('./summary') const { make_table, make_tg_table, make_html, summary } = require('./summary')
@ -18,7 +18,7 @@ const { https_proxy } = process.env
const axins = axios.create(https_proxy ? { httpsAgent: new HttpsProxyAgent(https_proxy) } : {}) const axins = axios.create(https_proxy ? { httpsAgent: new HttpsProxyAgent(https_proxy) } : {})
const SA_BATCH_SIZE = 1000 const SA_BATCH_SIZE = 1000
const SA_FILES = fs.readdirSync(path.join(__dirname, '../sa')).filter(v => v.endsWith('.json')) const SA_FILES = fs.readdirSync(path.join(__dirname, SA_Path)).filter(v => v.endsWith('.json'))
SA_FILES.flag = 0 SA_FILES.flag = 0
let SA_TOKENS = get_sa_batch() let SA_TOKENS = get_sa_batch()