这是一个扩展, 不知道怎么编辑简书, 复制过来的不好看, 大家直接复制出去就行了, 主要看方法
@implementation UIViewController (IMYPublic)
- (UINavigationController*)imy_navigationController
{
UINavigationController* nav = nil;
if ([self isKindOfClass:[UINavigationController class]]) {
nav = (id)self;
} else {
if ([self isKindOfClass:[UITabBarController class]]) {
nav = [((UITabBarController*)self).selectedViewController imy_navigationController];
} else {
nav = self.navigationController;
}
}
return nav;
}
@end
