iptv-org-mirror/epg
1
0
Fork 0
mirror of https://github.com/iptv-org/epg.git synced 2025-05-10 00:50:09 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update create-databasee.js

Browse source
This commit is contained in:
Aleksandr Statciuk 2022-01-21 17:30:14 +03:00
parent 465d38abb9
commit af69100bc5
3 changed files with 24 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

32
scripts/commands/create-database.js
View file

@ -1,6 +1,6 @@
const { db, file, parser, logger } = require('../core')
const { program } = require('commander')
const _ = require('lodash')
const { shuffle } = require('lodash')
const options = program
.option(
@ -27,7 +27,8 @@ main()
async function getChannels() {
logger.info(`Loading channels...`)
const channels = []
let channels = {}
const files = await file.list(options.channels)
for (const filepath of files) {
const dir = file.dirname(filepath)
@ -37,17 +38,28 @@ async function getChannels() {
const configPath = `${dir}/${site}.config.js`
const config = require(file.resolve(configPath))
if (config.ignore) continue
const [__, gid] = filename.match(/_([a-z-]+)\.channels\.xml/i) || [null, null]
const [__, groupId] = filename.match(/_([a-z-]+)\.channels\.xml/i) || [null, null]
const items = await parser.parseChannels(filepath)
for (const item of items) {
const countryCode = item.xmltv_id.split('.')[1]
item.country = countryCode ? countryCode.toUpperCase() : null
item.channelsPath = filepath
item.configPath = configPath
item.gid = gid
channels.push(item)
const key = `${item.site}:${item.site_id}`
if (!channels[key]) {
const countryCode = item.xmltv_id.split('.')[1]
item.country = countryCode ? countryCode.toUpperCase() : null
item.channelsPath = filepath
item.configPath = configPath
item.groups = []
channels[key] = item
}
if (!channels[key].groups.includes(groupId)) {
channels[key].groups.push(groupId)
}
}
}
channels = Object.values(channels)
logger.info(`Found ${channels.length} channels`)
return channels
@ -57,7 +69,7 @@ async function saveToDatabase(channels = []) {
logger.info('Saving to the database...')
await db.channels.load()
await db.channels.reset()
const chunks = split(_.shuffle(channels), options.maxClusters)
const chunks = split(shuffle(channels), options.maxClusters)
for (const [i, chunk] of chunks.entries()) {
for (const item of chunk) {
item.cluster_id = i + 1