def get_mark(bookId):
res = ''
is_print_all = input('选择：所有标注/指定章节的标注(1/2)?\n')
if is_print_all.strip() == '1': # 如果选择输出所有标注
way_to_print_chapter = input('选择：包含所有标题/只包含有标注的标题(1/2)?\n')
if way_to_print_chapter.strip() == '1': # 如果选择输出所有标题
res = get_bookmarklist(bookId)
elif way_to_print_chapter.strip() == '2': # 如果选择只输出包含标注的标题
res = get_bookmarklist(bookId, is_all_chapter=0)
else:
print('其他指令，默认包含所有标题\n')
res = get_bookmarklist(bookId)
elif is_print_all.strip() == '2': # 如果选择输出指定章节的标注
# 输出章节id和章节名
print_chapterUid_and_title(bookId)
# 接收章节id并去除空符
chapterUid_str = input('输入章节ID,多个ID用英文逗号分隔：\n').replace(' ', '')
print('获取中...')
# 遍历所有章节id
for chapterUid_s in chapterUid_str.split(','):
chapterUid = int(chapterUid_s)
res += get_bookmarklist(bookId,
is_all_chapter=1,
chapterUid=chapterUid)
print(chapterUid)
print('获取结束')
else:
print('其他指令，默认返回所有标注\n')
res = get_bookmarklist(bookId)
# 没有标注时给出提示
if res.strip() == '':
print('无标注')
return res
def main():
print('欢迎使用微信读书爬虫')
try:
books = get_bookshelf()
except:
print('请检查您的Cookie设置')
return
print('您的书架如下:')
for b in books:
print(b.bookId,b.title,b.author)
while True:
bid = input('想看哪本书的目录，请输入对应书籍id:')
for c in get_chapters(int(bid)):
print('#'*c[0],c[1])
y = input('是否需要查看该书的热门划线，y/n?')
if y.lower() == 'y':
bb = get_bestbookmarks(bid)
print(bb)
y = input('是否需要查看你的笔记，y/n?')
if y.lower() == 'y':
bb = get_bookmarklist(bid)
print(bb)
y = input('是否需要查看其他书，y/n?')
if y.lower() == 'y':
continue
else:
print('bye~')
break
def get_note(self, id):
note_name = os.path.join(self.note_dir, "%s.md" % id)
if os.path.exists(note_name):
with open(note_name, "r", encoding="utf-8") as f:
return f.read()
else:
return wereader.get_bookmarklist(id, cookies=self.cookies)
def export_mine():
global CURRENT
bid = CURRENT.bookId
bb = get_bookmarklist(bid, cookies=cookies)
with open(f"{CURRENT.title}-{CURRENT.bookId}.{FMT}", "w") as f:
f.write(bb)
print("导出成功")
def export_mine():
global CURRENT
bid = CURRENT.bookId
bb = get_bookmarklist(bid)
with open(f'{CURRENT.title}-{CURRENT.bookId}.{FMT}', 'w') as f:
f.write(bb)
print("导出成功")
def old_main():
print('欢迎使用微信读书爬虫')
try:
books = get_bookshelf()
except:
print('请检查您的Cookie设置')
return
print('您的书架如下:')
for b in books:
print(b.bookId, b.title, b.author)
while True:
bid = input('想看哪本书的目录，请输入对应书籍id:')
current_book = [x for x in books if x.bookId == bid][0]
for c in get_chapters(int(bid)):
print('#' * c[0], c[1])
y = input('是否需要查看该书的热门划线，y/n?')
if y.lower() == 'y':
bb = get_bestbookmarks(bid)
print(bb)
y = input('是否需要查看你的笔记，y/n?')
if y.lower() == 'y':
bb = get_bookmarklist(bid)
print(bb)
y = input('是否需要保存你的笔记，y/n?')
if y.lower() == 'y':
bb = get_bookmarklist(bid)
with open(f'{current_book.title}-{current_book.bookId}.txt',
'w') as f:
f.write(bb)
y = input('是否需要查看其他书，y/n?')
if y.lower() == 'y':
continue
else:
print('bye~')
break
def see_mine():
global CURRENT
bid = CURRENT.bookId
bb = get_bookmarklist(bid, cookies=cookies)
print(bb)
