twilight-library/app/test.hs

71 lines
2.4 KiB
Haskell

{-# LANGUAGE Arrows, NoMonomorphismRestriction #-}
import ShadowLibrary.Core
import Text.XML.HXT.Core
import Text.XML.HXT.XPath
import Data.List
import Data.List.Utils (replace)
import Text.Regex.Posix
import Text.Printf
import System.IO
extractYears = extractLinksWithText "//div[@class='inner clearfix']//div[@class='row']//div[@class='col-sm-6']//ul//li//a" -- pary adres-tytuł
runYearsExtractor url = runX $ (arr (const url) >>> setTraceLevel 1 >>> extractYears)
extractNewFiles = extractLinksWithText "//a[@title='Pobierz plik']" -- pary adres-tytuł
runNewFilesExtractor url = runX $ (arr (const url) >>> setTraceLevel 1 >>> extractNewFiles)
extractOldFiles = extractLinksWithText "//div[@class='content']//div[@class='inner clearfix']//ul//li//a" -- pary adres-tytuł
runOldFilesExtractor url = runX $ (arr (const url) >>> setTraceLevel 1 >>> extractOldFiles)
--367-376
extractOldYears = extractLinksWithText "//nav[@class='nav-left']//ul//li[@class='single-article']/a" -- pary adres-tytuł
runOldYearsExtractor url = runX $ (arr (const url) >>> setTraceLevel 1 >>> extractOldYears)
mapToUrl :: ([Char], [Char]) -> [Char]
mapToUrl (url, title) = url
merge [] ys = ys
merge (x:xs) ys = x:merge ys xs
repeatNTimes 0 = return []
repeatNTimes n = do
let link = "https://bip.um.gdynia.pl/module/Contents/controller/Default/action/loadSubcategory/subcategory_id/" ++ show (377-n)
items <- runOldYearsExtractor link
let urls = map mapToUrl items
recursive <- repeatNTimes (n-1)
let results = merge urls recursive
return results
flatten :: [[a]] -> [a]
flatten xs = (\z n -> foldr (\x y -> foldr z y x) n xs) (:) []
getAllPages url = do
items <- runYearsExtractor url
let urls = map mapToUrl items
newFiles <- mapM runNewFilesExtractor urls
let fileUrls = map mapToUrl (flatten newFiles)
return fileUrls
getOldFiles = do
oldPages <- repeatNTimes 10
--let sub = take 2 oldPages
oldFiles <- mapM runOldFilesExtractor oldPages
let fileUrls = map mapToUrl (flatten oldFiles)
return fileUrls
main = do
let start = "https://bip.um.gdynia.pl/wyszukiwarka-uchwal-rady-miasta,7485/wyszukiwarka-uchwal,504923"
oldResults <- getOldFiles
oldF <- openFile "oldFiles.json" WriteMode
hPrint oldF oldResults
hClose oldF
--print (length results)
newResults <- getAllPages start
newF <- openFile "newFiles.json" WriteMode
hPrint newF newResults
hClose newF