-
Notifications
You must be signed in to change notification settings - Fork 0
/
HsPrefix.py
40 lines (32 loc) · 1.01 KB
/
HsPrefix.py
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
#!/usr/bin/env python
"""
Methods to select/validate HitSpool file prefixes
"""
from __future__ import print_function
class HsPrefix(object):
"""
Old file prefix logic
"""
ANON = "ANON"
HESE = "HESE"
SNALERT = "SNALERT"
LIVE = "i3live"
__LIST = (HESE, SNALERT, ANON, LIVE)
@classmethod
def guess_from_dir(cls, copydir):
if copydir is not None:
if 'hese' in copydir:
return cls.HESE
elif 'HsDataCopy' in copydir:
return cls.SNALERT
elif 'HitSpool' in copydir and ('satellite' in copydir or
'archive' in copydir):
return cls.LIVE
return cls.ANON
@classmethod
def is_valid(cls, prefix):
return prefix is not None and prefix in cls.__LIST
if __name__ == "__main__":
for pfx in (HsPrefix.ANON, HsPrefix.HESE, HsPrefix.SNALERT, HsPrefix.LIVE,
"XXX"):
print("%s valid? == %s" % (pfx, HsPrefix.is_valid(pfx)))