Add parsing to cli
This commit is contained in:
@@ -1,10 +1,9 @@
|
|||||||
import argparse
|
import argparse
|
||||||
from datetime import datetime
|
|
||||||
import yaml
|
import yaml
|
||||||
from pydantic import ValidationError
|
from pydantic import ValidationError
|
||||||
|
|
||||||
from beaky.config import Config
|
from beaky.config import Config
|
||||||
from beaky.scanner.scanner import Link
|
from beaky.scanner.scanner import Links
|
||||||
from beaky.screenshotter.screenshotter import Screenshotter
|
from beaky.screenshotter.screenshotter import Screenshotter
|
||||||
|
|
||||||
|
|
||||||
@@ -28,11 +27,18 @@ def main() -> None:
|
|||||||
if config is None:
|
if config is None:
|
||||||
return
|
return
|
||||||
|
|
||||||
|
# always load testing data, we will modify that later
|
||||||
|
data = Links(config)
|
||||||
|
data.ret_links()
|
||||||
|
link_amount = len(data.links)
|
||||||
|
print(f"We found {link_amount} links")
|
||||||
|
if link_amount == 0:
|
||||||
|
print("ERROR, no links found")
|
||||||
|
return
|
||||||
|
|
||||||
if args.mode == "screenshotter":
|
if args.mode == "screenshotter":
|
||||||
screenshotter = Screenshotter(config)
|
screenshotter = Screenshotter(config)
|
||||||
screenshotter.capture_tickets([Link(1,
|
screenshotter.capture_tickets(data.links)
|
||||||
"https://applink.ifortuna.cz/ticketdetail?id=eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJzdWIiOiJDRkUwUUJNNktDMzQyUjAwIiwicHJmIjoiUFVCTElDIiwiaXNzIjoiYmV0c2xpcC1zZXJ2aWNlIiwiaWF0IjoxNzcyODc2NTk0fQ.QGiBJRINDsSVKQn3WKRa7XDql5wiLDOG8R7QKc2bD-0&source=SB&deeplink=ftncz%3A%2F%2Fbetslip-history%2Fdetail%3Fid%3DeyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJzdWIiOiJDRkUwUUJNNktDMzQyUjAwIiwicHJmIjoiUFVCTElDIiwiaXNzIjoiYmV0c2xpcC1zZXJ2aWNlIiwiaWF0IjoxNzcyODc2NTk0fQ.QGiBJRINDsSVKQn3WKRa7XDql5wiLDOG8R7QKc2bD-0%26source%3DSB",
|
|
||||||
datetime.now())])
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
|
|||||||
Reference in New Issue
Block a user