D2StatGenerator/main.py at master · insomnial/D2StatGenerator · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
from app.Director import Director
from app.bungiemanifest import DestinyManifest
from app.PgcrCollector import PGCRCollector
from app.Zipper import Zipper
from app.bungieapi import BungieApi
from app.reports.ActivityCountReport import ActivityCountReport
from app.reports.ActivityLocationTimeReport import ActivityLocationTimeReport
from app.reports.ActivityLocationWeaponReport import ActivityLocationWeaponReport
from app.reports.ActivityTypeRaceReport import ActivityTypeRaceReport
from app.reports.ActivityWinrateReport import ActivityWinrateReport
from app.reports.FireteamActivityReport import FireteamActivityReport
from app.reports.FireteamRace import FireteamRaceReport
from app.reports.KDReport import KDReport
from app.reports.KillsDeathsAssistsReport import KillsDeathsAssistsReport
from app.reports.LightLevelReport import LightLevelReport
from app.reports.PlaytimeCharacterReport import PlaytimeCharacterReport
from app.reports.PlaytimeReport import PlaytimeReport
from app.reports.WeaponKillTreeReport import WeaponKillTreeReport
from app.reports.WeaponRaceReport import WeaponRaceReport
from app.reports.WeaponReport import WeaponReport
from app.reports.WeekdayReport import WeekdayReport

###############################################################################
#
# main()
#
###############################################################################
if __name__ == '__main__':
    import pathos, argparse, os

    # build argument parsing
    descriptionString = """Get and compile stats for a Destiny 2 user.
    example: main.py -p 3 -id 4611686018472661350"""
    platformString = """    Xbox     1
    Psn      2
    Steam    3
    Blizzard 4
    Stadia   5
    Egs      6"""
    parser = argparse.ArgumentParser(prog='main.py', description=f'{descriptionString}', formatter_class=argparse.RawTextHelpFormatter)
    parser.add_argument('--platform', '-p', type=int, required=False, help=f'{platformString}')
    parser.add_argument('--membership_id', '-id', type=int, required=False, help='Bungie ID')
    parser.add_argument('--stale', '-s', action='store_true', required=False, help='Run in stale mode (no fresh PGCRs)')
    args = vars(parser.parse_args())
    platform = args['platform']
    id = args['membership_id']
    stale = args['stale']

    if platform != None and id != None:
        USED_MEMBERSHIP = (platform, id)
    else:
        MIJAGO = (3, 4611686018482684809)
        # You can easily set your own ID here:
        MYCOOLID = (3, 1234567890123456789)
        USED_MEMBERSHIP = MIJAGO

    from pathos.multiprocessing import ProcessPool, ThreadPool, ThreadingPool
    pathos.helpers.freeze_support()  # required for windows
    pool = ProcessPool()
    # You could also specify the amount of threads. Note that this DRASTICALLY speeds up the process but takes serious computation power.
    # pool = ProcessPool(128)

    # check manifest
    manifest = DestinyManifest().update()

    # You can also set an api key manually, if you do not want to use environment variables.
    API_KEY = os.getenv('BUNGIE_API_KEY')
    # API_KEY = "123456789"

    api = BungieApi(API_KEY)
    # "mp4" if you installed ffmpeg which you should; see README.d. otherwise "gif" if you do not.
    VIDEO_TYPE = "mp4"

    pc = PGCRCollector(*USED_MEMBERSHIP, api, pool)
    displayName = pc.getProfile().getDisplayName()

    if not stale:
        Director.CreateDirectoriesForUser(displayName)
        Director.ClearResultDirectory(displayName)
        Director.CreateDirectoriesForUser(displayName)

        pc.getCharacters().getActivities(limit=None).getPGCRs()  # .combineAllPgcrs()
        data = pc.getAllPgcrs()

        pool.close()

    reports = [
        ActivityCountReport(*USED_MEMBERSHIP, displayName, manifest),
        ActivityLocationTimeReport(*USED_MEMBERSHIP, displayName, manifest),
        ActivityLocationWeaponReport(*USED_MEMBERSHIP, displayName, manifest),
        ActivityTypeRaceReport(*USED_MEMBERSHIP, displayName, manifest, video_type=VIDEO_TYPE),
        ActivityWinrateReport(*USED_MEMBERSHIP, displayName, manifest),
        FireteamActivityReport(*USED_MEMBERSHIP, displayName, manifest),
        FireteamRaceReport(*USED_MEMBERSHIP, displayName, manifest, video_type=VIDEO_TYPE),
        KDReport(*USED_MEMBERSHIP, displayName, manifest),
        KillsDeathsAssistsReport(*USED_MEMBERSHIP, displayName, manifest),
        LightLevelReport(*USED_MEMBERSHIP, displayName, manifest),
        PlaytimeCharacterReport(*USED_MEMBERSHIP, displayName, manifest),
        PlaytimeReport(*USED_MEMBERSHIP, displayName, manifest),
        WeaponKillTreeReport(*USED_MEMBERSHIP, displayName, manifest),
        WeaponRaceReport(*USED_MEMBERSHIP, displayName, manifest, video_type=VIDEO_TYPE),
        WeaponReport(*USED_MEMBERSHIP, displayName, manifest),
        WeekdayReport(*USED_MEMBERSHIP, displayName, manifest)
    ]
    for report in reports:
        report.generate(data).save()

    Zipper.zip_directory(Director.GetResultDirectory(displayName), Director.GetZipPath(displayName))
    print("Generated ZIP:", Director.GetZipPath(displayName))