jiarongqiu commited on
Commit
e17abc6
·
1 Parent(s): 886047c
Files changed (2) hide show
  1. service/dataset.py +0 -58
  2. service/logger.py +0 -54
service/dataset.py DELETED
@@ -1,58 +0,0 @@
1
- import os
2
- import json
3
- from typing import Any
4
- import pytz
5
- from pathlib import Path
6
- from huggingface_hub import CommitScheduler
7
- from datetime import datetime
8
- from datasets import load_dataset
9
-
10
-
11
- class Dataset:
12
-
13
- DIR = os.path.join(os.path.dirname(__file__),"..","data")
14
-
15
- def __init__(self):
16
- self.scheduler = CommitScheduler(
17
- repo_id="Jarvis-Log",
18
- repo_type="dataset",
19
- folder_path=self.DIR,
20
- path_in_repo="data",
21
- token=os.getenv("HF_TOKEN")
22
- )
23
-
24
- def current_time(self):
25
- beijing_tz = pytz.timezone('Asia/Shanghai')
26
- beijing_time = datetime.now(beijing_tz)
27
- format_time = beijing_time.strftime('%Y-%m-%d %H:%M:%S')
28
- return format_time
29
-
30
- def current_date(self):
31
- beijing_tz = pytz.timezone('Asia/Shanghai')
32
- beijing_time = datetime.now(beijing_tz)
33
- format_date = beijing_time.strftime('%Y-%m-%d')
34
- return format_date
35
-
36
- def _write_log(self, data, dir_name):
37
- path = Path(self.DIR,dir_name)
38
- path.mkdir(parents=True, exist_ok=True)
39
- path = path/f'{self.current_date()}.json'
40
- data['datetime'] = self.current_time()
41
- data = json.dumps(data)+"\n"
42
- with self.scheduler.lock:
43
- with path.open("a") as f:
44
- f.write(data)
45
-
46
- def write_dataset(self,data):
47
- self._write_log(data,"dataset")
48
-
49
- def write_log(self,data):
50
- self._write_log(data,"log")
51
-
52
- def __call__(self,data):
53
- print(f"[{self.current_time()}]{data}")
54
-
55
- def load_dataset(self):
56
- dataset = load_dataset("JR818/Jarvis-Log",data_dir="data/dataset",ignore_verifications=True)
57
- return dataset.with_format("pandas").data
58
- dataset = Dataset()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
service/logger.py DELETED
@@ -1,54 +0,0 @@
1
- import os
2
- import json
3
- from typing import Any
4
- import pytz
5
- from pathlib import Path
6
- from huggingface_hub import CommitScheduler
7
- from datetime import datetime
8
-
9
-
10
- class Logger:
11
-
12
- DIR = os.path.join(os.path.dirname(__file__),"..","data")
13
-
14
- def __init__(self):
15
- self.scheduler = CommitScheduler(
16
- repo_id="Jarvis-Log",
17
- repo_type="dataset",
18
- folder_path=self.DIR,
19
- path_in_repo="data",
20
- token=os.getenv("HF_TOKEN")
21
- )
22
-
23
- def current_time(self):
24
- beijing_tz = pytz.timezone('Asia/Shanghai')
25
- beijing_time = datetime.now(beijing_tz)
26
- format_time = beijing_time.strftime('%Y-%m-%d %H:%M:%S')
27
- return format_time
28
-
29
- def current_date(self):
30
- beijing_tz = pytz.timezone('Asia/Shanghai')
31
- beijing_time = datetime.now(beijing_tz)
32
- format_date = beijing_time.strftime('%Y-%m-%d')
33
- return format_date
34
-
35
- def _write_log(self, data, dir_name):
36
- path = Path(self.DIR,dir_name)
37
- path.mkdir(parents=True, exist_ok=True)
38
- path = path/f'{self.current_date()}.json'
39
- data['datetime'] = self.current_time()
40
- data = json.dumps(data)+"\n"
41
- with self.scheduler.lock:
42
- with path.open("a") as f:
43
- f.write(data)
44
-
45
- def write_dataset(self,data):
46
- self._write_log(data,"dataset")
47
-
48
- def write_log(self,data):
49
- self._write_log(data,"log")
50
-
51
- def __call__(self,data):
52
- print(f"[{self.current_time()}]{data}")
53
-
54
- logger = Logger()